GoLear是一款Go语言机器学习框架,示例代码:
package maiimport ( "fmt" "github.com/sjwhitworth/golear/base" "github.com/sjwhitworth/golear/evaluatio" "github.com/sjwhitworth/golear/k")fuc mai() { // Load i a dataset, with headers. Header attributes will be stored. // Thik of istaces as a Data Frame structure i R or Padas. // You ca also create istaces from scratch. rawData, err := base.ParseCSVToIstaces("datasets/iris.csv", false) if err != il { paic(err) } // Prit a pleasat summary of your data. fmt.Pritl(rawData) //Iitialises a ew KNN classifier cls := k.NewKClassifier("euclidea", 2) //Do a traiig-test split traiData, testData := base.IstacesTraiTestSplit(rawData, 0.50) cls.Fit(traiData) //Calculates the Euclidea distace ad returs the most popular label predictios := cls.Predict(testData) fmt.Pritl(predictios) // Prits precisio/recall metrics cofusioMat, err := evaluatio.GetCofusioMatrix(testData, predictios) if err != il { paic(fmt.Spritf("Uable to get cofusio matrix: %s", err.Error())) } fmt.Pritl(evaluatio.GetSummary(cofusioMat))}点击空白处退出提示
评论