brain is a personal experimental project that under heavy development, backward api compatibility is not guaranteed!
data, target := datasets.LoadIris()
clf := knn.KNeighborsClassifier(5)
clf.Fit(data, target)
log.Println(clf.Predict([]float64{5.9, 3, 5.1, 1.8}))
listOfPosts, listClasses := datasets.LoadSpam()
vocabList := datasets.CreateVocabList(listOfPosts)
inputVec := datasets.SetOfWords2Vec(vocabList, listOfPosts[2])
nb := BernoulliNB()
var trainMat [][]float64
for _, v := range listOfPosts {
trainMat = append(trainMat, datasets.SetOfWords2Vec(vocabList, v))
}
nb.Fit(trainMat, listClasses)
t.Log(nb.Predict(inputVec))
t.Log(nb.Measure(trainMat, listClasses))
- knn
- naive bayes
- svm
- tree
- neural network