mirror of
https://github.com/sjwhitworth/golearn.git
synced 2025-04-26 13:49:14 +08:00
1809a8b358
- BaseClassifier Predict and Fit methods return errors - go fmt ./... Conflicts: ensemble/randomforest.go ensemble/randomforest_test.go trees/tree_test.go
31 lines
803 B
Go
31 lines
803 B
Go
package base
|
|
|
|
import (
|
|
"github.com/gonum/matrix/mat64"
|
|
)
|
|
|
|
// Classifier implementations predict categorical class labels.
|
|
type Classifier interface {
|
|
// Takes a set of Instances, copies the class Attribute
|
|
// and constructs a new set of Instances of equivalent
|
|
// length with only the class Attribute and fills it in
|
|
// with predictions.
|
|
Predict(FixedDataGrid) (FixedDataGrid, error)
|
|
// Takes a set of instances and updates the Classifier's
|
|
// internal structures to enable prediction
|
|
Fit(FixedDataGrid) error
|
|
// Why not make every classifier return a nice-looking string?
|
|
String() string
|
|
}
|
|
|
|
// BaseClassifier stores options common to every classifier.
|
|
type BaseClassifier struct {
|
|
TrainingData *DataGrid
|
|
}
|
|
|
|
type BaseRegressor struct {
|
|
Data mat64.Dense
|
|
Name string
|
|
Labels []float64
|
|
}
|