Package org.languagetool.rules.spelling.suggestions

Class XGBoostSuggestionsOrderer

java.lang.Object

org.languagetool.rules.spelling.suggestions.SuggestionsOrdererFeatureExtractor

org.languagetool.rules.spelling.suggestions.XGBoostSuggestionsOrderer

All Implemented Interfaces:

SuggestionsOrderer, SuggestionsRanker

public class XGBoostSuggestionsOrderer
extends SuggestionsOrdererFeatureExtractor
implements SuggestionsRanker

Field Summary

Fields inherited from class org.languagetool.rules.spelling.suggestions.SuggestionsOrdererFeatureExtractor

language, languageModel, mistakeProb, score, topN

Constructor Summary

Constructors

Constructor	Description
XGBoostSuggestionsOrderer(Language lang, LanguageModel languageModel)

Method Summary

Modifier and Type	Method	Description
protected void	initParameters()
boolean	isMlAvailable()
List<SuggestedReplacement>	orderSuggestions(List<String> suggestions, String word, AnalyzedSentence sentence, int startPos)
static void	setAutoCorrectThresholdForLanguage(Language lang, float value)	For testing purposes only
boolean	shouldAutoCorrect(List<SuggestedReplacement> rankedSuggestions)	Model output should have been calibrated using a precision-recall curve evaluation, so that a threshold for confidence values with sufficiently high precision for auto correction is known

Methods inherited from class org.languagetool.rules.spelling.suggestions.SuggestionsOrdererFeatureExtractor

computeFeatures

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.languagetool.rules.spelling.suggestions.SuggestionsOrderer

orderSuggestionsUsingModel

Constructor Details

XGBoostSuggestionsOrderer

public XGBoostSuggestionsOrderer(Language lang,
 LanguageModel languageModel)

Method Details

setAutoCorrectThresholdForLanguage

public static void setAutoCorrectThresholdForLanguage(Language lang,
 float value)

For testing purposes only

initParameters

protected void initParameters()

Overrides:

initParameters in class SuggestionsOrdererFeatureExtractor

isMlAvailable

public boolean isMlAvailable()

Specified by:

isMlAvailable in interface SuggestionsOrderer

Overrides:

isMlAvailable in class SuggestionsOrdererFeatureExtractor

orderSuggestions

public List<SuggestedReplacement> orderSuggestions(List<String> suggestions,
 String word,
 AnalyzedSentence sentence,
 int startPos)

Specified by:

orderSuggestions in interface SuggestionsOrderer

Overrides:

orderSuggestions in class SuggestionsOrdererFeatureExtractor

shouldAutoCorrect

public boolean shouldAutoCorrect(List<SuggestedReplacement> rankedSuggestions)

Description copied from interface: SuggestionsRanker

Model output should have been calibrated using a precision-recall curve evaluation, so that a threshold for confidence values with sufficiently high precision for auto correction is known

Specified by:

shouldAutoCorrect in interface SuggestionsRanker

Parameters:

rankedSuggestions - suggestions returned by orderSuggestions

Returns:

if confidence is high enough for auto correction