summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authordesaicwtf <desaicwtf@ec762483-ff6d-05da-a07a-a48fb63a330f>2010-07-23 17:23:13 +0000
committerdesaicwtf <desaicwtf@ec762483-ff6d-05da-a07a-a48fb63a330f>2010-07-23 17:23:13 +0000
commit58681ee5816d13c04002ca8aebe23c2768da4e5b (patch)
tree71115a0dd70d4ea2328d2b806234d895399a5666
parent76ef39de737e7abc0a8fe989dfacb7885617e59f (diff)
git-svn-id: https://ws10smt.googlecode.com/svn/trunk@381 ec762483-ff6d-05da-a07a-a48fb63a330f
-rw-r--r--gi/posterior-regularisation/prjava/src/phrase/Trainer.java4
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;