Commit 4ab0962b authored by Sebastian N.'s avatar Sebastian N.
Browse files

Added perplexity as eval metric

parent 00c9d6c3
Pipeline #192291 failed with stages
in 25 seconds
......@@ -41,6 +41,7 @@ grammar CNNTrain extends de.monticore.lang.monticar.Common2, de.monticore.Number
| f1:"f1"
| mae:"mae"
| mse:"mse"
| perplexity:"perplexity"
| rmse:"rmse"
| topKAccuracy:"top_k_accuracy");
......
......@@ -44,6 +44,12 @@ public enum EvalMetric {
return "mse";
}
},
PERPLEXITY{
@Override
public String toString() {
return "perplexity";
}
},
RMSE{
@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