Commit 00c9d6c3 authored by Sebastian N.'s avatar Sebastian N.
Browse files

Added BLEU as eval metric

parent 5555a601
Pipeline #192287 passed with stages
in 9 minutes and 16 seconds
......@@ -18,7 +18,7 @@
<groupId>de.monticore.lang.monticar</groupId>
<artifactId>cnn-train</artifactId>
<version>0.3.7-SNAPSHOT</version>
<version>0.3.8-SNAPSHOT</version>
<!-- == PROJECT DEPENDENCIES ============================================= -->
......
......@@ -36,6 +36,7 @@ grammar CNNTrain extends de.monticore.lang.monticar.Common2, de.monticore.Number
LossWeightsEntry implements ConfigEntry = name:"loss_weights" ":" value:DoubleVectorValue;
EvalMetricValue implements ConfigValue =(accuracy:"accuracy"
| bleu:"bleu"
| crossEntropy:"cross_entropy"
| f1:"f1"
| mae:"mae"
......
......@@ -14,6 +14,12 @@ public enum EvalMetric {
return "accuracy";
}
},
BLEU{
@Override
public String toString() {
return "bleu";
}
},
CROSS_ENTROPY{
@Override
public String toString() {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment