Versions in this module Expand all Collapse all v1 v1.0.3 Feb 13, 2024 v1.0.1 Feb 7, 2024 Changes in this version + func ParameterGrid(paramGrid map[string][]interface{}) (out []map[string]interface{}) + func TrainTestSplit(X, Y mat.Matrix, testsize float64, randomstate uint64) (Xtrain, Xtest, ytrain, ytest *mat.Dense) + type CrossValidateResult struct + Estimator []base.Predicter + FitTime []time.Duration + ScoreTime []time.Duration + TestScore []float64 + func CrossValidate(estimator base.Predicter, X, Y *mat.Dense, groups []int, ...) (res CrossValidateResult) + func (r CrossValidateResult) Len() int + func (r CrossValidateResult) Less(i, j int) bool + func (r CrossValidateResult) Swap(i, j int) + type GridSearchCV struct + BestEstimator base.Predicter + BestIndex int + BestParams map[string]interface{} + BestScore float64 + CV Splitter + CVResults map[string][]interface{} + Estimator base.Predicter + LowerScoreIsBetter bool + NJobs int + NOutputs int + ParamGrid map[string][]interface{} + RandomState rand.Source + Scorer func(Ytrue, Ypred mat.Matrix) float64 + UseChannels bool + Verbose bool + func (gscv *GridSearchCV) Fit(Xmatrix, Ymatrix mat.Matrix) base.Fiter + func (gscv *GridSearchCV) GetNOutputs() int + func (gscv *GridSearchCV) IsClassifier() bool + func (gscv *GridSearchCV) Predict(X mat.Matrix, Y mat.Mutable) *mat.Dense + func (gscv *GridSearchCV) PredicterClone() base.Predicter + func (gscv *GridSearchCV) Score(X, Y mat.Matrix) float64 + type KFold struct + NSplits int + RandomState base.RandomState + Shuffle bool + func (splitter *KFold) GetNSplits(X, Y *mat.Dense) int + func (splitter *KFold) Split(X, Y *mat.Dense) (ch chan Split) + func (splitter *KFold) SplitterClone() Splitter + type RandomState = rand.Rand + type Split struct + TestIndex []int + TrainIndex []int + type Splitter interface + GetNSplits func(X, Y *mat.Dense) int + Split func(X, Y *mat.Dense) (ch chan Split) + SplitterClone func() Splitter
Go to main page