Commitce5b31e

committed

add test function

1 parentc7dce73 commitce5b31eCopy full SHA for ce5b31e

File tree

-1

lines changed

-1

lines changed

Lines changed: 2 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -9,11 +9,12 @@`
`9`	`9`
`10`	`10`	`classRECNN(RECNNBase):`
`11`	`11`	`def__init__(self,config:RECNNConfig,dtype:type=tf.float32,dict_path:str='',mode:str='train',`
`12`		`-data_path:str='',relation_count:int=2):`
	`12`	`+data_path:str='',relation_count:int=2,model_path:str=''):`
`13`	`13`	`RECNNBase.__init__(self,config,dict_path)`
`14`	`14`	`self.dtype=dtype`
`15`	`15`	`self.mode=mode`
`16`	`16`	`self.data_path=data_path`
	`17`	`+self.model_path=model_path`
`17`	`18`	`self.relation_count=relation_count`
`18`	`19`
`19`	`20`	`self.concat_embed_size=self.word_embed_size+2*self.position_embed_size`

Lines changed: 2 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -9,5 +9,7 @@ def train_re_cnn():`
`9`	`9`	`recnn=RECNN(config=config,data_path=data_path,dict_path=dict_path)`
`10`	`10`	`recnn.fit()`
`11`	`11`
	`12`	`+deftest_re_cnn():`
	`13`	`+pass`
`12`	`14`	`if__name__=='__main__':`
`13`	`15`	`train_re_cnn()`

Comments

(0)