diff options
author | desaicwtf <desaicwtf@ec762483-ff6d-05da-a07a-a48fb63a330f> | 2010-07-23 17:23:13 +0000 |
---|---|---|
committer | desaicwtf <desaicwtf@ec762483-ff6d-05da-a07a-a48fb63a330f> | 2010-07-23 17:23:13 +0000 |
commit | 6c26dfd1e1cc1974c8d7840d825a77bf55464bef (patch) | |
tree | a1b067c676ae0a4bcf624e5e7d0926d2a655f1c2 /gi/posterior-regularisation | |
parent | 0f6b65e81a655e5935fd7260abb554f3a54d94d1 (diff) |
git-svn-id: https://ws10smt.googlecode.com/svn/trunk@381 ec762483-ff6d-05da-a07a-a48fb63a330f
Diffstat (limited to 'gi/posterior-regularisation')
-rw-r--r-- | gi/posterior-regularisation/prjava/src/phrase/Trainer.java | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/gi/posterior-regularisation/prjava/src/phrase/Trainer.java b/gi/posterior-regularisation/prjava/src/phrase/Trainer.java index cea6a20a..f205ce67 100644 --- a/gi/posterior-regularisation/prjava/src/phrase/Trainer.java +++ b/gi/posterior-regularisation/prjava/src/phrase/Trainer.java @@ -36,7 +36,7 @@ public class Trainer parser.accepts("convergence-threshold").withRequiredArg().ofType(Double.class).defaultsTo(1e-6); parser.accepts("variational-bayes"); parser.accepts("alpha-emit").withRequiredArg().ofType(Double.class).defaultsTo(0.1); - parser.accepts("alpha-pi").withRequiredArg().ofType(Double.class).defaultsTo(0.01); + parser.accepts("alpha-pi").withRequiredArg().ofType(Double.class).defaultsTo(0.0001); parser.accepts("agree-direction"); parser.accepts("agree-language"); parser.accepts("no-parameter-cache"); @@ -120,6 +120,8 @@ public class Trainer if (vb) { //cluster.initialiseVB(alphaEmit, alphaPi); vbModel=new VB(tags,corpus); + vbModel.alpha=alphaPi; + vbModel.lambda=alphaEmit; } if (options.has("no-parameter-cache")) cluster.cacheLambda = false; |