diff options
| author | trevor.cohn <trevor.cohn@ec762483-ff6d-05da-a07a-a48fb63a330f> | 2010-07-21 22:48:02 +0000 |
|---|---|---|
| committer | trevor.cohn <trevor.cohn@ec762483-ff6d-05da-a07a-a48fb63a330f> | 2010-07-21 22:48:02 +0000 |
| commit | e5f89e61768c7a8cc3ac60976d14314de60862f4 (patch) | |
| tree | 9d2e932aeee368daf7de0e5dcd38f5cf1a5d5418 /gi/posterior-regularisation/prjava/src | |
| parent | 9e3c1c2e7ee29d8967dab8e0d4d321fdccd5624b (diff) | |
Posterior output
git-svn-id: https://ws10smt.googlecode.com/svn/trunk@366 ec762483-ff6d-05da-a07a-a48fb63a330f
Diffstat (limited to 'gi/posterior-regularisation/prjava/src')
| -rw-r--r-- | gi/posterior-regularisation/prjava/src/phrase/Agree.java | 7 | ||||
| -rw-r--r-- | gi/posterior-regularisation/prjava/src/phrase/Trainer.java | 12 |
2 files changed, 11 insertions, 8 deletions
diff --git a/gi/posterior-regularisation/prjava/src/phrase/Agree.java b/gi/posterior-regularisation/prjava/src/phrase/Agree.java index f3dc5c36..8f7b499e 100644 --- a/gi/posterior-regularisation/prjava/src/phrase/Agree.java +++ b/gi/posterior-regularisation/prjava/src/phrase/Agree.java @@ -182,7 +182,12 @@ public class Agree { public void displayPosterior(PrintStream ps)
{
- for (Edge edge : c.getEdges())
+ displayPosterior(ps, c.getEdges());
+ }
+
+ public void displayPosterior(PrintStream ps, List<Edge> test)
+ {
+ for (Edge edge : test)
{
double probs[] = posterior(edge);
arr.F.l1normalize(probs);
diff --git a/gi/posterior-regularisation/prjava/src/phrase/Trainer.java b/gi/posterior-regularisation/prjava/src/phrase/Trainer.java index c1d4775e..b51db919 100644 --- a/gi/posterior-regularisation/prjava/src/phrase/Trainer.java +++ b/gi/posterior-regularisation/prjava/src/phrase/Trainer.java @@ -180,23 +180,21 @@ public class Trainer File outfile = (File) options.valueOf("out"); try { PrintStream ps = FileUtil.printstream(outfile); - List<Edge> test; - if (!options.has("test")) // just use the training - test = corpus.getEdges(); - else + List<Edge> test = corpus.getEdges(); + if (options.has("test")) // just use the training { // if --test supplied, load up the file - if (agree == null && agree2sides == null) + if (agree2sides == null) { infile = (File) options.valueOf("test"); System.out.println("Reading testing concordance from " + infile); test = corpus.readEdges(FileUtil.reader(infile)); } else - System.err.println("Can't run agreement models on different test data cf training (yet); --test ignored."); + System.err.println("Can't run bilingual agreement model on different test data cf training (yet); --test ignored."); } if (agree != null) - agree.displayPosterior(ps); + agree.displayPosterior(ps, test); else if (agree2sides != null) agree2sides.displayPosterior(ps); else |