Package ai.djl.fasttext.jni

Class FtWrapper

java.lang.Object

ai.djl.util.NativeResource<Long>

ai.djl.fasttext.jni.FtWrapper

All Implemented Interfaces:

AutoCloseable

public final class FtWrapper
extends ai.djl.util.NativeResource<Long>

A class containing utilities to interact with the fastText JNI layer.

Field Summary

Fields inherited from class ai.djl.util.NativeResource

handle

Method Summary

Modifier and Type	Method	Description
boolean	checkModel(String modelFilePath)
void	close()
String	getModelType()
float[]	getWordVector(String word)
void	loadModel(String modelFilePath)
static FtWrapper	newInstance()
ai.djl.modality.Classifications	predictProba(String text, int topK, String labelPrefix)
void	runCmd(String[] args)
void	unloadModel()

Methods inherited from class ai.djl.util.NativeResource

getHandle, getUid, isReleased, onClose

Methods inherited from class java.lang.Object

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

Method Details

newInstance

public static FtWrapper newInstance()

loadModel

public void loadModel(String modelFilePath)

checkModel

public boolean checkModel(String modelFilePath)

unloadModel

public void unloadModel()

getModelType

public String getModelType()

predictProba

public ai.djl.modality.Classifications predictProba(String text,
 int topK,
 String labelPrefix)

getWordVector

public float[] getWordVector(String word)

runCmd

public void runCmd(String[] args)

close

public void close()

Specified by:

close in interface AutoCloseable

Overrides:

close in class ai.djl.util.NativeResource<Long>