Skip to content

Commit 844806a

Browse files
committed
evaluation: Minor fix
1 parent e3f474e commit 844806a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dynaml-core/src/main/scala-2.11/io/github/mandar2812/dynaml/evaluation/ClassificationMetricsTF.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,6 @@ class ClassificationMetricsTF(num_classes: Int, preds: Tensor, targets: Tensor)
5252
println()
5353

5454
scala.Predef.print("Class Prediction Fidelity Score: ")
55-
pprint.pprintln(class_score.scalar.asInstanceOf[Double])
55+
pprint.pprintln(class_score.scalar.asInstanceOf[Float])
5656
}
5757
}

0 commit comments

Comments
 (0)