luckyimnotanigger/HeliBoard
1
0
Fork 0
mirror of https://github.com/Helium314/HeliBoard.git synced 2025-04-25 09:06:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

tune language weights for multilingual typing

Browse source
This commit is contained in:
Helium314 2023-11-22 22:41:41 +01:00
parent 38fde35890
commit 48e0b05d26
1 changed files with 19 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
app/src/main/java/org/dslul/openboard/inputmethod/latin/DictionaryFacilitatorImpl.java
View file

@ -131,7 +131,6 @@ public class DictionaryFacilitatorImpl implements DictionaryFacilitator {
*/ */
private static class DictionaryGroup { private static class DictionaryGroup {
private static final int MAX_CONFIDENCE = 2; private static final int MAX_CONFIDENCE = 2;
private static final int MIN_CONFIDENCE = 0;
/** /**
* The locale associated with the dictionary group. * The locale associated with the dictionary group.
@ -158,8 +157,6 @@ public class DictionaryFacilitatorImpl implements DictionaryFacilitator {
// when decreasing confidence or getting weight factor, limit to maximum // when decreasing confidence or getting weight factor, limit to maximum
public void increaseConfidence() { public void increaseConfidence() {
mConfidence += 1; mConfidence += 1;
if (mConfidence <= MAX_CONFIDENCE)
updateWeights();
} }
// If confidence is above max, drop to max confidence. This does not change weights and // If confidence is above max, drop to max confidence. This does not change weights and
@ -167,20 +164,28 @@ public class DictionaryFacilitatorImpl implements DictionaryFacilitator {
public void decreaseConfidence() { public void decreaseConfidence() {
if (mConfidence > MAX_CONFIDENCE) if (mConfidence > MAX_CONFIDENCE)
mConfidence = MAX_CONFIDENCE; mConfidence = MAX_CONFIDENCE;
else if (mConfidence > MIN_CONFIDENCE) { else if (mConfidence > 0) {
mConfidence -= 1; mConfidence -= 1;
updateWeights();
} }
} }
// todo: might need some more tuning, maybe more confidence steps public float getWeightForTypingInLocale(List<DictionaryGroup> groups) {
private void updateWeights() { return getWeightForLocale(groups, 0.15f);
mWeightForTypingInLocale = 1f - 0.15f * (MAX_CONFIDENCE - mConfidence);
mWeightForGesturingInLocale = 1f - 0.05f * (MAX_CONFIDENCE - mConfidence);
} }
public float mWeightForTypingInLocale = 1f; public float getWeightForGesturingInLocale(List<DictionaryGroup> groups) {
public float mWeightForGesturingInLocale = 1f; return getWeightForLocale(groups, 0.05f);
}
// might need some more tuning
private float getWeightForLocale(final List<DictionaryGroup> groups, final float step) {
if (groups.size() == 1) return 1f;
if (mConfidence < 2) return 1f - step * (MAX_CONFIDENCE - mConfidence);
for (DictionaryGroup group : groups) {
if (group != this && group.mConfidence >= mConfidence) return 1f - step / 2f;
}
return 1f;
}
public final ConcurrentHashMap<String, ExpandableBinaryDictionary> mSubDictMap = public final ConcurrentHashMap<String, ExpandableBinaryDictionary> mSubDictMap =
new ConcurrentHashMap<>(); new ConcurrentHashMap<>();
@ -876,9 +881,9 @@ public class DictionaryFacilitatorImpl implements DictionaryFacilitator {
int sessionId, long proximityInfoHandle, float[] weightOfLangModelVsSpatialModel, int sessionId, long proximityInfoHandle, float[] weightOfLangModelVsSpatialModel,
DictionaryGroup dictGroup) { DictionaryGroup dictGroup) {
final ArrayList<SuggestedWordInfo> suggestions = new ArrayList<>(); final ArrayList<SuggestedWordInfo> suggestions = new ArrayList<>();
final float weightForLocale = composedData.mIsBatchMode float weightForLocale = composedData.mIsBatchMode
? dictGroup.mWeightForGesturingInLocale ? dictGroup.getWeightForGesturingInLocale(mDictionaryGroups)
: dictGroup.mWeightForTypingInLocale; : dictGroup.getWeightForTypingInLocale(mDictionaryGroups);
for (final String dictType : ALL_DICTIONARY_TYPES) { for (final String dictType : ALL_DICTIONARY_TYPES) {
final Dictionary dictionary = dictGroup.getDict(dictType); final Dictionary dictionary = dictGroup.getDict(dictType);
if (null == dictionary) continue; if (null == dictionary) continue;