Package ai.djl.fasttext

Class TrainFastText

java.lang.Object

ai.djl.fasttext.TrainFastText

public final class TrainFastText
extends Object

A utility to aggregate options for training with fasttext.

Method Summary

Modifier and Type	Method	Description
static FtTextClassification	textClassification(FtTrainingConfig config, ai.djl.training.dataset.RawDataset<Path> dataset)	Trains a fastText Application.NLP.TEXT_CLASSIFICATION model.

Methods inherited from class java.lang.Object

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

Method Details

textClassification

public static FtTextClassification textClassification(FtTrainingConfig config,
 ai.djl.training.dataset.RawDataset<Path> dataset)
                                               throws IOException

Trains a fastText Application.NLP.TEXT_CLASSIFICATION model.

Parameters:

config - the training configuration to use

dataset - the training dataset

Returns:

the result of the training

Throws:

IOException - when IO operation fails in loading a resource

See Also:

• FtTextClassification.fit(FtTrainingConfig, RawDataset)