OrgLion-ML/golearn
1
0
Fork 0
mirror of https://github.com/sjwhitworth/golearn.git synced 2025-04-28 13:48:56 +08:00
Code Issues Releases Wiki Activity

base: correct handling of class attributes in ParseCSVToTemplatedInstances

Browse Source
This commit is contained in:
Richard Townsend 2016-07-11 23:16:18 +01:00
parent 457a00ac7c
commit 7041fc33c7
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
base/csv.go
View File

@ -303,6 +303,13 @@ func ParseCSVToTemplatedInstances(filepath string, hasHeaders bool, template *De
return nil, err
}
for _, a := range template.AllClassAttributes() {
err = instances.AddClassAttribute(a)
if err != nil {
return nil, err
}
}
return instances, nil
}

2
knn/knn_test.go
View File

@ -70,7 +70,7 @@ func TestKnnClassifierWithOptimisationsPartiallyComplete(t *testing.T) {
Convey("Given two basically identical files...", t, func() {
trainingData, err := base.ParseCSVToInstances("knn_train_2.csv", true)
So(err, ShouldBeNil)
testingData, err := base.ParseCSVToInstances("knn_test_2.csv", true)
testingData, err := base.ParseCSVToTemplatedInstances("knn_test_2.csv", true, trainingData)
So(err, ShouldBeNil)
cls := NewKnnClassifier("euclidean", 2)