1 files changed, 60 insertions, 0 deletions
diff --git a/gi/posterior-regularisation/prjava/src/test/CorpusTest.java b/gi/posterior-regularisation/prjava/src/test/CorpusTest.java
new file mode 100644
index 00000000..b4c3041f
--- /dev/null
+++ b/gi/posterior-regularisation/prjava/src/test/CorpusTest.java
@@ -0,0 +1,60 @@
+package test;
+
+import java.util.Arrays;
+import java.util.HashMap;
+
+import data.Corpus;
+import hmm.POS;
+
+public class CorpusTest {
+
+	public static void main(String[] args) {
+		Corpus c=new Corpus(POS.trainFilename);
+
+		
+		int idx=30;
+		
+		
+		HashMap<String, Integer>vocab=
+			(HashMap<String, Integer>) io.SerializedObjects.readSerializedObject(Corpus.alphaFilename);
+		
+		HashMap<String, Integer>tagVocab=
+			(HashMap<String, Integer>) io.SerializedObjects.readSerializedObject(Corpus.tagalphaFilename);
+		
+		
+		String [] dict=new String [vocab.size()+1];
+		for(String key:vocab.keySet()){
+			dict[vocab.get(key)]=key;
+		}
+		dict[dict.length-1]=Corpus.UNK_TOK;
+		
+		String [] tagdict=new String [tagVocab.size()+1];
+		for(String key:tagVocab.keySet()){
+			tagdict[tagVocab.get(key)]=key;
+		}
+		tagdict[tagdict.length-1]=Corpus.UNK_TOK;
+		
+		String[] sent=c.get(idx);
+		int []data=c.getInt(idx);
+		
+		
+		String []roundtrip=new String [sent.length];
+		for(int i=0;i<sent.length;i++){
+			roundtrip[i]=dict[data[i]];
+		}
+		System.out.println(Arrays.toString(sent));
+		System.out.println(Arrays.toString(roundtrip));
+		
+		sent=c.tag.get(idx);
+		data=c.tagData.get(idx);
+		
+		
+		roundtrip=new String [sent.length];
+		for(int i=0;i<sent.length;i++){
+			roundtrip[i]=tagdict[data[i]];
+		}
+		System.out.println(Arrays.toString(sent));
+		System.out.println(Arrays.toString(roundtrip));
+	}
+
+}