Changeset 33698 for other-projects/maori-lang-detection/src/org/greenstone/atea/morphia/SentenceInfo.java

Timestamp:

2019-11-15T23:14:48+13:00 (4 years ago)

Author:

ak19

Message:

Links to more reading

File:

• other-projects/maori-lang-detection/src/org/greenstone/atea/morphia/SentenceInfo.java (modified) (2 diffs)

other-projects/maori-lang-detection/src/org/greenstone/atea/morphia/SentenceInfo.java

@@ -4 +4 @@
 import java.util.HashMap;
 
+// morphia is used to map Java objects to mongodb
 import dev.morphia.annotations.*;
 
@@ -11 +12 @@
 
     public final String sentence;
-    public final Map<String, Double> languageToConfidenceMap;
-    @Embedded
-    public final LanguageInfo[] languagesInfo; // array of langCode and confidence value pairs
-
+    //public final Map<String, Double> languageToConfidenceMap;
+    public final String bestLangCode;
+    public final double bestLangConfidence;    
+    //@Embedded
+    //public final LanguageInfo[] languagesInfo; // array of langCode and confidence value pairs
+    
     
     public SentenceInfo(String sentence, LanguageInfo[] languages) {
-    this.sentence = sentence;
-    this.languagesInfo = languages;
+    this.sentence = sentence;   
+    if(languages.length >= 1) {
+        this.bestLangCode = languages[0].langCode;
+        this.bestLangConfidence = languages[0].confidence;
+    } else {
+        this.bestLangCode = null;
+        this.bestLangConfidence = 0.0;
+    }
 
+    //this.languagesInfo = null;
+    //this.languagesInfo = languages;
+
+    
     // let's store (langCode -> confidence) lookup in Map:
-    this.languageToConfidenceMap = new HashMap<String, Double>();   
+    //this.languageToConfidenceMap = new HashMap<String, Double>();
+    /*
     for(LanguageInfo li : languages) {
         String langCode = li.langCode;
-        Double confidence = new Double(li.confidenceLevel);
+        Double confidence = new Double(li.confidence);
         languageToConfidenceMap.put(langCode, confidence);
     }
+    */
     }
     
 }
 
-// BACK WHEN WE ONLY STORED THE BEST PREDICTED LANGUAGE META FOR EACH SENTENCE:
-/*
-@Entity("Sentences")
-public class SentenceInfo {
-    public final double confidenceLevel;
-    // 3 letter lang code
-    public final String langCode;
-    public final String sentence;
-    
-    public SentenceInfo(double confidence, String langCode, String sentence) {
-    this.confidenceLevel = confidence;
-    this.langCode = langCode;
-    this.sentence = sentence;
-    }
-}
-*/