com.datumbox.framework.core.machinelearning.featureselection

Class TFIDF

java.lang.Object

com.datumbox.framework.core.machinelearning.common.abstracts.AbstractTrainer<MP,TP>

com.datumbox.framework.core.machinelearning.common.abstracts.featureselectors.AbstractFeatureSelector<MP,TP>

com.datumbox.framework.core.machinelearning.common.abstracts.featureselectors.AbstractScoreBasedFeatureSelector<TFIDF.ModelParameters,TFIDF.TrainingParameters>

com.datumbox.framework.core.machinelearning.featureselection.TFIDF

All Implemented Interfaces:

Savable, Parallelizable, Trainable<TFIDF.ModelParameters,TFIDF.TrainingParameters>, java.lang.AutoCloseable

public class TFIDF
extends AbstractScoreBasedFeatureSelector<TFIDF.ModelParameters,TFIDF.TrainingParameters>

Implementation of the TF-IDF Feature Selection algorithm. * References: http://en.wikipedia.org/wiki/Tf%E2%80%93idf https://gist.github.com/AloneRoad/1605037 http://www.tfidf.com/

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	TFIDF.ModelParameters Base class for every ModelParameter class in the framework.
static class	TFIDF.TrainingParameters The AbstractTrainingParameters class stores the parameters that can be changed before training the algorithm.

Nested classes/interfaces inherited from class com.datumbox.framework.core.machinelearning.common.abstracts.featureselectors.AbstractScoreBasedFeatureSelector

AbstractScoreBasedFeatureSelector.AbstractModelParameters, AbstractScoreBasedFeatureSelector.AbstractTrainingParameters

Method Summary

Modifier and Type	Method and Description
void	fit(Dataframe trainingData) Trains a model using the provided training parameters and data.

Methods inherited from class com.datumbox.framework.core.machinelearning.common.abstracts.featureselectors.AbstractFeatureSelector

fit_transform, isParallelized, setParallelized, transform

Methods inherited from class com.datumbox.framework.core.machinelearning.common.abstracts.AbstractTrainer

close, delete, getModelParameters, getTrainingParameters, save

Methods inherited from class java.lang.Object

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

Method Detail

fit

public void fit(Dataframe trainingData)

Trains a model using the provided training parameters and data.

Specified by:

fit in interface Trainable<TFIDF.ModelParameters,TFIDF.TrainingParameters>

Overrides:

fit in class AbstractFeatureSelector<TFIDF.ModelParameters,TFIDF.TrainingParameters>