mirror of
https://github.com/sjwhitworth/golearn.git
synced 2025-05-03 22:17:14 +08:00
0035dd184e
This is the first draft of the bernoulli naive bayes implementation. It is missing the Fit function tests and the Predict function.
25 lines
508 B
Go
25 lines
508 B
Go
package naive
|
|
|
|
import (
|
|
"testing"
|
|
"github.com/gonum/matrix/mat64"
|
|
)
|
|
|
|
// Test if panic is correctly called when matrices with different
|
|
// dimensions are used as arguments.
|
|
func TestFitPanic(t *testing.T) {
|
|
defer func() {
|
|
if recover() == nil {
|
|
t.Fatalf("invalid matrix dim did not panic")
|
|
}
|
|
}()
|
|
|
|
nb := NewBernoulliNBClassifier(2)
|
|
|
|
X := mat64.NewDense(10, 20, nil)
|
|
// simulating user mistake, one extra label
|
|
y := make([]int, 11)
|
|
|
|
nb.Fit(X, y)
|
|
}
|