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