mlmetrics/accuracy_test.go at main · bsm/mlmetrics · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package mlmetrics_test

import (
	. "github.com/bsm/ginkgo"
	. "github.com/bsm/gomega"
	"github.com/bsm/mlmetrics"
)

var _ = Describe("Accuracy", func() {
	var subject *mlmetrics.Accuracy

	BeforeEach(func() {
		subject = mlmetrics.NewAccuracy()
		subject.Observe(1, 1)
		subject.Observe(1, 1)
		subject.Observe(1, 0)
		subject.Observe(0, 0)
		subject.Observe(0, 0)
		subject.Observe(0, 1)
		subject.Observe(1, 1)
		subject.ObserveWeight(1, 1, 1.0)
		subject.ObserveWeight(1, 1, 2.0)
		subject.ObserveWeight(0, 0, 1.0)
		subject.ObserveWeight(0, 1, 1.0)
	})

	It("should calculate stats", func() {
		Expect(subject.CorrectWeight()).To(Equal(9.0))
		Expect(subject.TotalWeight()).To(Equal(12.0))
		Expect(subject.Rate()).To(Equal(0.75))
	})

	It("should ignore non-categories", func() {
		subject.Observe(-1, 1)
		subject.Observe(1, -1)
		subject.Observe(-1, -1)
		Expect(subject.TotalWeight()).To(Equal(12.0))
	})
})