Skip to content
bits of sklearn ported to Go #golang
Branch: master
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
base remove most of staticcheck warnings May 14, 2019
cluster remove most of staticcheck warnings May 14, 2019
datasets remove most of staticcheck warnings May 14, 2019
interpolate adapted to gonum.org/v1/gonum@v0.0.0-20190331200053-3d26580ed485 Apr 2, 2019
linear_model remove most of staticcheck warnings May 14, 2019
metrics remove most of staticcheck warnings May 14, 2019
model_selection remove most of staticcheck warnings May 14, 2019
neighbors remove most of staticcheck warnings May 14, 2019
neural_network remove most of staticcheck warnings May 14, 2019
pipeline - improved coverage Apr 15, 2019
preprocessing added PowerTransformer (yeo-johnson and box-cox) May 15, 2019
svm remove most of staticcheck warnings May 14, 2019
.gitignore MLPClassifier/Regressor Unmarshal allow to reload a MLP saved from sc… Apr 7, 2019
.travis.yml travis: removed go1.11 and go-tip May 1, 2019
AUTHORS added AUTHORS CONTRIBUTORS and CONTRIBUTING.md Jun 12, 2018
CONTRIBUTING.md added AUTHORS CONTRIBUTORS and CONTRIBUTING.md Jun 12, 2018
CONTRIBUTORS
GOPHER
LICENSE Create LICENSE Feb 21, 2018
README.md updated preprocessing example list in README May 15, 2019
doc.go added doc.go files and twitter link in readme Apr 22, 2019
go.mod updated dependencies May 15, 2019
go.sum updated preprocessing example list in README May 15, 2019
gopher.png changed gopher image, and go versions in travis Apr 4, 2019

README.md

sklearn

Partial port of scikit-learn to go

License MIT Build Status Code Coverage Go Report Card GoDoc twitter

Examples

cluster

DBSCAN KMeans

datasets

LoadIris LoadBreastCancer LoadDiabetes LoadBoston LoadExamScore LoadMicroChipTest LoadMnist LoadMnistWeights MakeRegression MakeBlobs

interpolate

CubicSpline Interp1d Interp2d

linear_model

LinearRegression BayesianRidge MultiTaskElasticNet MultiTaskLasso ElasticNet Lasso LassoPath LogisticRegression Ridge

metrics

AccuracyScore ConfusionMatrix PrecisionScore RecallScore F1Score FBetaScore PrecisionRecallFScoreSupport ROCCurve AUC ROCAUCScore PrecisionRecallCurve AveragePrecisionScore R2Score

model_selection

KFold CrossValidate

neighbors

KNeighborsClassifier MinkowskiDistance EuclideanDistance KDTree NearestCentroid KNeighborsRegressor NearestNeighbors NearestNeighbors.KNeighborsGraph NearestNeighbors.Tree

neural_network

MLPClassifier.Unmarshal MLPClassifier.Fit.mnist MLPClassifier.Predict.mnist MLPClassifier.Fit.breast.cancer MLPRegressor.Fit.boston

pipeline

Pipeline

preprocessing

MinMaxScaler StandardScaler RobustScaler AddDummyFeature OneHotEncoder Shuffler MaxAbsScaler Binarizer Normalizer Scale KernelCenterer QuantileTransformer PowerTransformer PowerTransformer.boxcox KBinsDiscretizer FunctionTransformer Imputer LabelBinarizer MultiLabelBinarizer LabelEncoder PCA

svm

SVC SVR

This is a personal project to get a deeper understanding of how all of this magic works

  • linted with gofmt, golint, go vet revive

  • unit tested but coverage should reach 90%

  • underdocumented but scikit-learn doc is your friend

Many thanks to gonum and scikit-learn authors and contributors

PRs are welcome

You can’t perform that action at this time.