diff --git a/examples/knnclassifier/knnclassifier_iris.go b/examples/knnclassifier/knnclassifier_iris.go index 592e3ee..0661ce9 100644 --- a/examples/knnclassifier/knnclassifier_iris.go +++ b/examples/knnclassifier/knnclassifier_iris.go @@ -15,7 +15,7 @@ func main() { } //Initialises a new KNN classifier - cls := knn.NewKnnClassifier("euclidean", 2) + cls := knn.NewKnnClassifier("euclidean", "linear", 2) //Do a training-test split trainData, testData := base.InstancesTrainTestSplit(rawData, 0.50)