/
LanguageIdentificationActivity.kt
73 lines (63 loc) · 2.62 KB
/
LanguageIdentificationActivity.kt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package com.google.example.mlkit.kotlin
import android.os.Bundle
import android.util.Log
import androidx.appcompat.app.AppCompatActivity
import com.google.mlkit.nl.languageid.LanguageIdentification
import com.google.mlkit.nl.languageid.LanguageIdentificationOptions
private const val TAG = "LangIDActivity"
class LanguageIdentificationActivity : AppCompatActivity() {
override fun onCreate(savedInstanceState: Bundle?) {
super.onCreate(savedInstanceState)
}
fun identifyLanguageWithStringInput(text: String) {
// [START identify_languages]
val languageIdentifier = LanguageIdentification.getClient()
languageIdentifier.identifyLanguage(text)
.addOnSuccessListener { languageCode ->
if (languageCode == "und") {
Log.i(TAG, "Can't identify language.")
} else {
Log.i(TAG, "Language: $languageCode")
}
}
.addOnFailureListener {
// Model couldn’t be loaded or other internal error.
// ...
}
// [END identify_languages]
}
fun setConfidence() {
// [START set_confidence]
val languageIdentifier = LanguageIdentification
.getClient(LanguageIdentificationOptions.Builder()
.setConfidenceThreshold(0.34f)
.build())
// [END set_confidence]
}
fun getPossibleLanguuages(text: String) {
// [START get_possible_languages]
val languageIdentifier = LanguageIdentification.getClient()
languageIdentifier.identifyPossibleLanguages(text)
.addOnSuccessListener { identifiedLanguages ->
for (identifiedLanguage in identifiedLanguages) {
val language = identifiedLanguage.languageTag
val confidence = identifiedLanguage.confidence
Log.i(TAG, "$language $confidence")
}
}
.addOnFailureListener {
// Model couldn’t be loaded or other internal error.
// ...
}
// [END get_possible_languages]
}
private fun setConfidenceThreshold() {
// [START set_confidence_threshold]
val identifierOptions = LanguageIdentificationOptions.Builder()
.setConfidenceThreshold(0.5f)
.build()
val languageIdentifier = LanguageIdentification
.getClient(identifierOptions)
// [END set_confidence_threshold]
}
}