public class TrainSectorAnnotator extends Object
| Modifier and Type | Class and Description |
|---|---|
protected static class |
TrainSectorAnnotator.ExecParams |
| Modifier and Type | Field and Description |
|---|---|
protected static org.slf4j.Logger |
log |
| Constructor and Description |
|---|
TrainSectorAnnotator() |
| Modifier and Type | Method and Description |
|---|---|
protected SectorAnnotator.Builder |
initializeClassLabelsTarget(SectorAnnotator.Builder builder,
Dataset train)
Initialize target encoder for the class labels task (single-label classification)
|
protected SectorAnnotator.Builder |
initializeHeadingsTarget(SectorAnnotator.Builder builder,
Dataset train,
WordHelpers.Language lang)
Initialize target encoder for the headings prediction task (multi-label classification)
|
protected SectorAnnotator.Builder |
initializeInputEncodings_bloom(SectorAnnotator.Builder builder,
Dataset train,
WordHelpers.Language lang) |
protected SectorAnnotator.Builder |
initializeInputEncodings_wemb(SectorAnnotator.Builder builder,
Resource embeddingModel) |
static void |
main(String[] args) |
protected void |
runTraining(TrainSectorAnnotator.ExecParams params) |
public static void main(String[] args) throws IOException
IOExceptionprotected void runTraining(TrainSectorAnnotator.ExecParams params) throws IOException
IOExceptionprotected SectorAnnotator.Builder initializeInputEncodings_bloom(SectorAnnotator.Builder builder, Dataset train, WordHelpers.Language lang)
protected SectorAnnotator.Builder initializeInputEncodings_wemb(SectorAnnotator.Builder builder, Resource embeddingModel) throws IOException
IOExceptionprotected SectorAnnotator.Builder initializeClassLabelsTarget(SectorAnnotator.Builder builder, Dataset train)
protected SectorAnnotator.Builder initializeHeadingsTarget(SectorAnnotator.Builder builder, Dataset train, WordHelpers.Language lang)
Copyright © 2019. All rights reserved.