Commit90891ae

committed

modify hyperparameter according to [XingChi Chen 2015]

1 parent798cadf commit90891aeCopy full SHA for 90891ae

File tree

-4

lines changed

-4

lines changed

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -2,8 +2,8 @@`
`2`	`2`
`3`	`3`
`4`	`4`	`classDnnCrfConfig(object):`
`5`		`-def__init__(self,*,skip_left:int=1,skip_right:int=0,embed_size:int=200,hidden_units:int=150,`
`6`		`-learning_rate:float=0.01,lam:float=1e-4,dropout_rate:float=0.6,batch_length:int=300,`
	`5`	`+def__init__(self,*,skip_left:int=0,skip_right:int=2,embed_size:int=100,hidden_units:int=150,`
	`6`	`+learning_rate:float=0.2,lam:float=1e-4,dropout_rate:float=0.2,batch_length:int=300,`
`7`	`7`	`batch_size=20):`
`8`	`8`	`self.__skip_left=skip_left`
`9`	`9`	`self.__skip_right=skip_right`

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,7 @@`
`10`	`10`	`deftrain_cws():`
`11`	`11`	`data_path='../dnlp/data/cws/pku_training.pickle'`
`12`	`12`	`config=DnnCrfConfig()`
`13`		`-dnncrf=DnnCrf(config=config,data_path=data_path,nn='bilstm')`
	`13`	`+dnncrf=DnnCrf(config=config,data_path=data_path,nn='lstm')`
`14`	`14`	`dnncrf.fit()`
`15`	`15`
`16`	`16`
`@@ -19,7 +19,7 @@ def test_cws():`
`19`	`19`	`sentence='中国人民决心继承邓小平同志的遗志，继续把建设有中国特色社会主义事业推向前进。'`
`20`	`20`	`model_path='../dnlp/models/cws32.ckpt'`
`21`	`21`	`config=DnnCrfConfig()`
`22`		`-dnncrf=DnnCrf(config=config,mode='predict',model_path=model_path,nn='bilstm')`
	`22`	`+dnncrf=DnnCrf(config=config,mode='predict',model_path=model_path,nn='lstm')`
`23`	`23`	`res,labels=dnncrf.predict_ll(sentence,return_labels=True)`
`24`	`24`	`print(res)`
`25`	`25`	`evaluate_cws(dnncrf,'../dnlp/data/cws/pku_test.pickle')`

Comments

(0)