Categorygithub.com/pa-m/sklearnmodel_selection
package
0.0.0-20200711083454-beb861ee48b1
Repository: https://github.com/pa-m/sklearn.git
Documentation: pkg.go.dev
Versions
1
Dependencies
10
Dependents
0
Files
440 SLOC

# Functions

CrossValidate Evaluate a score by cross-validation scorer is a func(Ytrue,Ypred) float64 only mean_squared_error for now NJobs is the number of goroutines.
ParameterGrid ...
TrainTestSplit splits X and Y into test set and train set testsize must be between 0 and 1 it produce same sets than scikit-learn.

# Structs

CrossValidateResult is the struct result of CrossValidate.
GridSearchCV ..
KFold ...
Split ...

# Interfaces

Splitter is the interface for splitters like KFold.

# Type aliases

RandomState is to init a new random source for reproducibility.