golem ~master (2021-09-25T16:54:00Z)
Dub
Repo
confusionMatrix
golem
metrics
Undocumented in source. Be warned that the author may not have intended to support it.
Tensor
!(
size_t
, [
Shape
[1],
Shape
[1]],
UseGradient.no
)
confusionMatrix
(
T
size_t
[]
Shape
UseGradient
useGrad1
UseGradient
useGrad2
)
(
Tensor
!(
T
,
Shape
,
useGrad1
)
x
,
Tensor
!(
T
,
Shape
,
useGrad2
)
y
)
if
(
Shape.length
== 2 &&
Shape
[1] > 1
)
Meta
Source
See Implementation
golem
metrics
functions
accuracy
confusionMatrix