public class LabelSeeker extends Object
| Constructor and Description |
|---|
LabelSeeker(List<String> labelsUsed,
org.deeplearning4j.models.embeddings.inmemory.InMemoryLookupTable<org.deeplearning4j.models.word2vec.VocabWord> lookupTable) |
| Modifier and Type | Method and Description |
|---|---|
List<org.nd4j.linalg.primitives.Pair<String,Double>> |
getScores(org.nd4j.linalg.api.ndarray.INDArray vector)
This method accepts vector, that represents any document,
and returns distances between this document, and previously trained categories
|
org.nd4j.linalg.api.ndarray.INDArray |
getScoresAsVector(org.nd4j.linalg.api.ndarray.INDArray vector) |
public List<org.nd4j.linalg.primitives.Pair<String,Double>> getScores(org.nd4j.linalg.api.ndarray.INDArray vector)
public org.nd4j.linalg.api.ndarray.INDArray getScoresAsVector(org.nd4j.linalg.api.ndarray.INDArray vector)
Copyright © 2019. All rights reserved.