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

Rebasing against master

Browse Source
This commit is contained in:
Richard Townsend 2015-01-15 22:53:44 +00:00
parent 5c302a11ea
commit 04e6c8a414
3 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/averageperceptron/averageperceptionexample.go
View File

@ -29,6 +29,6 @@ func main() {
predictions := cls.Predict(testData)
// Prints precision/recall metrics
confusionMat := evaluation.GetConfusionMatrix(testData, predictions)
confusionMat, _ := evaluation.GetConfusionMatrix(testData, predictions)
fmt.Println(evaluation.GetSummary(confusionMat))
}

2
perceptron/average.go
View File

@ -106,7 +106,7 @@ func (p *AveragePerceptron) Predict(what base.FixedDataGrid) base.FixedDataGrid
data := processData(what)
allAttrs := base.CheckCompatable(what, p.TrainingData)
allAttrs := base.CheckCompatible(what, p.TrainingData)
if allAttrs == nil {
// Don't have the same Attributes
return nil

6
perceptron/average_test.go
View File

@ -76,7 +76,11 @@ func TestPredict(t *testing.T) {
}
predictions := a.Predict(testData)
cf := evaluation.GetConfusionMatrix(testData, predictions)
cf, err := evaluation.GetConfusionMatrix(testData, predictions)
if err != nil {
t.Errorf("Couldn't get confusion matrix: %s", err)
t.Fail()
}
fmt.Println(evaluation.GetSummary(cf))
fmt.Println(trainData)
fmt.Println(testData)