public static class HeadingTag.Factory extends Object
| Modifier and Type | Field and Description |
|---|---|
protected BagOfWordsEncoder |
encoder |
| Constructor and Description |
|---|
Factory(BagOfWordsEncoder encoder) |
| Modifier and Type | Method and Description |
|---|---|
void |
attachFromSectionAnnotations(Document doc,
Annotation.Source source) |
void |
attachFromSectionAnnotations(Document doc,
Annotation.Source source,
boolean attachVectors) |
void |
attachFromSentenceVectors(Document doc,
Class<? extends Encoder> encoder,
Annotation.Source source) |
HeadingTag |
create(org.nd4j.linalg.api.ndarray.INDArray prediction) |
HeadingTag |
create(String heading) |
protected final BagOfWordsEncoder encoder
public Factory(BagOfWordsEncoder encoder)
public HeadingTag create(String heading)
public HeadingTag create(org.nd4j.linalg.api.ndarray.INDArray prediction)
public void attachFromSectionAnnotations(Document doc, Annotation.Source source)
public void attachFromSectionAnnotations(Document doc, Annotation.Source source, boolean attachVectors)
public void attachFromSentenceVectors(Document doc, Class<? extends Encoder> encoder, Annotation.Source source)
Copyright © 2019. All rights reserved.