mirror of
https://github.com/sjwhitworth/golearn.git
synced 2025-04-28 13:48:56 +08:00
a2c67592df
* Refactors KNNClassifier to use them * csv handling moved back into base due to a circular dependency * Also adds the datasets used to test CSV handling
108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
package base
|
|
|
|
import "testing"
|
|
|
|
func isSortedAsc(inst *Instances, attrIndex int) bool {
|
|
valPrev := 0.0
|
|
for i := 0; i < inst.Rows; i++ {
|
|
cur := inst.Get(i, attrIndex)
|
|
if i > 0 {
|
|
if valPrev > cur {
|
|
return false
|
|
}
|
|
}
|
|
valPrev = cur
|
|
}
|
|
return true
|
|
}
|
|
|
|
func isSortedDesc(inst *Instances, attrIndex int) bool {
|
|
valPrev := 0.0
|
|
for i := 0; i < inst.Rows; i++ {
|
|
cur := inst.Get(i, attrIndex)
|
|
if i > 0 {
|
|
if valPrev < cur {
|
|
return false
|
|
}
|
|
}
|
|
valPrev = cur
|
|
}
|
|
return true
|
|
}
|
|
|
|
func TestSortDesc(testEnv *testing.T) {
|
|
inst1, err := ParseCSVToInstances("../examples/datasets/iris_headers.csv", true)
|
|
if err != nil {
|
|
testEnv.Error(err)
|
|
return
|
|
}
|
|
inst2, err := ParseCSVToInstances("../examples/datasets/iris_sorted_desc.csv", true)
|
|
if err != nil {
|
|
testEnv.Error(err)
|
|
return
|
|
}
|
|
|
|
if isSortedDesc(inst1, 0) {
|
|
testEnv.Error("Can't test descending sort order")
|
|
}
|
|
if !isSortedDesc(inst2, 0) {
|
|
testEnv.Error("Reference data not sorted in descending order!")
|
|
}
|
|
attrs := make([]int, 4)
|
|
attrs[0] = 3
|
|
attrs[1] = 2
|
|
attrs[2] = 1
|
|
attrs[3] = 0
|
|
inst1.Sort(Descending, attrs)
|
|
if !isSortedDesc(inst1, 0) {
|
|
testEnv.Error("Instances are not sorted in descending order")
|
|
testEnv.Error(inst1)
|
|
}
|
|
if !inst2.Equal(inst1) {
|
|
inst1.storage.Sub(inst1.storage, inst2.storage)
|
|
testEnv.Error(inst1.storage)
|
|
testEnv.Error("Instances don't match")
|
|
testEnv.Error(inst1)
|
|
testEnv.Error(inst2)
|
|
}
|
|
}
|
|
|
|
func TestSortAsc(testEnv *testing.T) {
|
|
inst, err := ParseCSVToInstances("../examples/datasets/iris_headers.csv", true)
|
|
if isSortedAsc(inst, 0) {
|
|
testEnv.Error("Can't test ascending sort on something ascending already")
|
|
}
|
|
if err != nil {
|
|
testEnv.Error(err)
|
|
return
|
|
}
|
|
attrs := make([]int, 4)
|
|
attrs[0] = 3
|
|
attrs[1] = 2
|
|
attrs[2] = 1
|
|
attrs[3] = 0
|
|
inst.Sort(Ascending, attrs)
|
|
if !isSortedAsc(inst, 0) {
|
|
testEnv.Error("Instances are not sorted in ascending order")
|
|
testEnv.Error(inst)
|
|
}
|
|
|
|
inst2, err := ParseCSVToInstances("../examples/datasets/iris_sorted_asc.csv", true)
|
|
if err != nil {
|
|
testEnv.Error(err)
|
|
return
|
|
}
|
|
if !isSortedAsc(inst2, 0) {
|
|
testEnv.Error("This file should be sorted in ascending order")
|
|
}
|
|
|
|
if !inst2.Equal(inst) {
|
|
inst.storage.Sub(inst.storage, inst2.storage)
|
|
testEnv.Error(inst.storage)
|
|
testEnv.Error("Instances don't match")
|
|
testEnv.Error(inst)
|
|
testEnv.Error(inst2)
|
|
}
|
|
|
|
}
|