mirror of
https://github.com/sjwhitworth/golearn.git
synced 2025-04-26 13:49:14 +08:00
31 lines
797 B
Go
31 lines
797 B
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
base "github.com/sjwhitworth/golearn/base"
|
|
evaluation "github.com/sjwhitworth/golearn/evaluation"
|
|
knn "github.com/sjwhitworth/golearn/knn"
|
|
)
|
|
|
|
func main() {
|
|
rawData, err := base.ParseCSVToInstances("../datasets/iris_headers.csv", true)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
//Initialises a new KNN classifier
|
|
cls := knn.NewKnnClassifier("euclidean", 2)
|
|
|
|
//Do a training-test split
|
|
trainData, testData := base.InstancesTrainTestSplit(rawData, 0.50)
|
|
cls.Fit(trainData)
|
|
|
|
//Calculates the Euclidean distance and returns the most popular label
|
|
predictions := cls.Predict(testData)
|
|
fmt.Println(predictions)
|
|
|
|
// Prints precision/recall metrics
|
|
confusionMat := evaluation.GetConfusionMatrix(testData, predictions)
|
|
fmt.Println(evaluation.GetSummary(confusionMat))
|
|
}
|