deeplearning4j / deeplearning4j/deeplearning4j-examples

how to predict in CNNTextClassifier.java?

Open
#587 3 comments 0 reactions 0 assignees View on GitHub
question
Dominant language
Java
Stars
2.5k
Forks
1.8k
PR merge metrics
No merged PRs in 30d

Description

1.the model use ReshapePreProcessor
```
graphConfBuilder.addVertex(reshapedForConvName,
new PreprocessorVertex(new ReshapePreProcessor(
new int[] {batchSize * docLength, embeddingsDim},
new int[]{batchSize, 1, docLength, embeddingsDim})),
embeddedName);
```
2. when predict,there is only one sample ,there will be a error "Mis matched lengths",how to fix the bug?

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.