Skip to content

Latest commit

 

History

105 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

go-ml-deployment

ChatGPT Image May 1, 2025, 09_40_12 AM

Project Objectives

  • Should be able to deploy sklearn models as a go-server
  • Production ready and well-tested

Supported Models

Name Package Sklearn-onnx Support Our Support
ADBRegression linear_model ✅ ❌
AdaBoostClassifier ensemble ✅ ❌
AdaBoostRegressor ensemble ✅ ❌
AdditiveChi2Sampler kernel_approximation ❌ ❌
AffinityPropagation cluster ❌ ❌
AgglomerativeClustering cluster ❌ ❌
BaggingClassifier ensemble ✅ ❌
BaggingRegressor ensemble ✅ ❌
BaseDecisionTree tree ✅ ❌
BaseEnsemble ensemble ❌ ❌
BayesianGaussianMixture mixture ✅ ❌
BayesianRidge linear_model ✅ ❌
BernoulliNB naive_bayes ✅ ❌
BernoulliRBM neural_network ❌ ❌
Binarizer preprocessing ✅ ❌
Birch cluster ❌ ❌
BisectingKMeans cluster ❌ ❌
CCA cross_decomposition ❌ ❌
CalibratedClassifierCV calibration ✅ ❌
CategoricalNB naive_bayes ✅ ❌
ClassifierChain multioutput ❌ ❌
ComplementNB naive_bayes ✅ ❌
DBSCAN cluster ❌ ❌
DecisionTreeClassifier tree ✅ ❌
DecisionTreeRegressor tree ✅ ❌
DictVectorizer feature_extraction ✅ ❌
DictionaryLearning decomposition ❌ ❌
ElasticNet linear_model ✅ ❌
ElasticNetCV linear_model ✅ ❌
EllipticEnvelope covariance ❌ ❌
EmpiricalCovariance covariance ❌ ❌
ExtraTreeClassifier tree ✅ ❌
ExtraTreeRegressor tree ✅ ❌
ExtraTreesClassifier ensemble ✅ ❌
ExtraTreesRegressor ensemble ✅ ❌
FactorAnalysis decomposition ❌ ❌
FastICA decomposition ❌ ❌
FeatureAgglomeration cluster ❌ ❌
FeatureHasher feature_extraction ✅ ❌
FixedThresholdClassifier model_selection ❌ ❌
FrozenEstimator frozen ❌ ❌
FunctionTransformer preprocessing ✅ ❌
GammaRegressor linear_model ✅ ❌
GaussianMixture mixture ✅ ❌
GaussianNB naive_bayes ✅ ❌
GaussianProcessClassifier gaussian_process ✅ ❌
GaussianProcessRegressor gaussian_process ✅ ❌
GaussianRandomProjection random_projection ✅ ❌
GenericUnivariateSelect feature_selection ✅ ❌
GradientBoostingClassifier ensemble ✅ ❌
GradientBoostingRegressor ensemble ✅ ❌
GraphicalLasso covariance ❌ ❌
GraphicalLassoCV covariance ❌ ❌
GridSearchCV model_selection ✅ ❌
HDBSCAN cluster ❌ ❌
HistGradientBoostingClassifier ensemble ✅ ❌
HistGradientBoostingRegressor ensemble ✅ ❌
HuberRegressor linear_model ✅ ❌
IncrementalPCA decomposition ✅ ❌
IsolationForest ensemble ✅ ❌
IsotonicRegression isotonic ❌ ❌
KBinsDiscretizer preprocessing ✅ ❌
KMeans cluster ✅ ❌
KNNImputer impute ✅ ❌
KNeighborsClassifier neighbors ✅ ❌
KNeighborsRegressor neighbors ✅ ❌
KNeighborsTransformer neighbors ✅ ❌
KernelCenterer preprocessing ✅ ❌
KernelDensity neighbors ❌ ❌
KernelPCA decomposition ✅ ❌
KernelRidge kernel_ridge ❌ ❌
LabelBinarizer preprocessing ✅ ❌
LabelEncoder preprocessing ✅ ❌
LabelPropagation semi_supervised ❌ ❌
LabelSpreading semi_supervised ❌ ❌
Lars linear_model ✅ ❌
LarsCV linear_model ✅ ❌
Lasso linear_model ✅ ❌
LassoCV linear_model ✅ ❌
LassoLars linear_model ✅ ❌
LassoLarsCV linear_model ✅ ❌
LassoLarsIC linear_model ✅ ❌
LatentDirichletAllocation decomposition ❌ ❌
LedoitWolf covariance ❌ ❌
LinearDiscriminantAnalysis discriminant_analysis ✅ ❌
LinearRegression linear_model ✅ ❌
LinearSVC svm ✅ ❌
LinearSVR svm ✅ ❌
LocalOutlierFactor neighbors ✅ ❌
LogisticRegression linear_model ✅ ❌
LogisticRegressionCV linear_model ✅ ❌
MLPClassifier neural_network ✅ ❌
MLPRegressor neural_network ✅ ❌
MaxAbsScaler preprocessing ✅ ❌
MeanShift cluster ❌ ❌
MinCovDet covariance ❌ ❌
MinMaxScaler preprocessing ✅ ❌
MiniBatchDictionaryLearning decomposition ❌ ❌
MiniBatchKMeans cluster ✅ ❌
MiniBatchNMF decomposition ❌ ❌
MiniBatchSparsePCA decomposition ❌ ❌
MissingIndicator impute ❌ ❌
MultiLabelBinarizer preprocessing ❌ ❌
MultiOutputClassifier multioutput ✅ ❌
MultiOutputRegressor multioutput ✅ ❌
MultiTaskElasticNet linear_model ✅ ❌
MultiTaskElasticNetCV linear_model ✅ ❌
MultiTaskLasso linear_model ✅ ❌
MultiTaskLassoCV linear_model ✅ ❌
MultinomialNB naive_bayes ✅ ❌
NMF decomposition ❌ ❌
NearestCentroid neighbors ❌ ❌
NearestNeighbors neighbors ✅ ❌
NeighborhoodComponentsAnalysis neighbors ✅ ❌
Normalizer preprocessing ✅ ❌
NuSVC svm ✅ ❌
NuSVR svm ✅ ❌
Nystroem kernel_approximation ❌ ❌
OAS covariance ❌ ❌
OPTICS cluster ❌ ❌
OneClassSVM svm ✅ ❌
OneHotEncoder preprocessing ✅ ❌
OneVsOneClassifier multiclass ✅ ❌
OneVsRestClassifier multiclass ✅ ❌
OrdinalEncoder preprocessing ✅ ❌
OrthogonalMatchingPursuit linear_model ✅ ❌
OrthogonalMatchingPursuitCV linear_model ✅ ❌
OutputCodeClassifier multiclass ❌ ❌
PCA decomposition ✅ ❌
PLSCanonical cross_decomposition ❌ ❌
PLSRegression cross_decomposition ✅ ❌
PLSSVD cross_decomposition ❌ ❌
PassiveAggressiveClassifier linear_model ✅ ❌
PassiveAggressiveRegressor linear_model ✅ ❌
Perceptron linear_model ✅ ❌
PoissonRegressor linear_model ✅ ❌
PolynomialCountSketch kernel_approximation ❌ ❌
PolynomialFeatures preprocessing ✅ ❌
PowerTransformer preprocessing ✅ ❌
QuadraticDiscriminantAnalysis discriminant_analysis ✅ ❌
QuantileRegressor linear_model ✅ ❌
QuantileTransformer preprocessing ❌ ❌
RANSACRegressor linear_model ✅ ❌
RBFSampler kernel_approximation ❌ ❌
RFE feature_selection ✅ ❌
RFECV feature_selection ✅ ❌
RadiusNeighborsClassifier neighbors ✅ ❌
RadiusNeighborsRegressor neighbors ✅ ❌
RadiusNeighborsTransformer neighbors ❌ ❌
RandomForestClassifier ensemble ✅ ❌
RandomForestRegressor ensemble ✅ ❌
RandomTreesEmbedding ensemble ✅ ❌
RandomizedSearchCV model_selection ❌ ❌
RegressorChain multioutput ❌ ❌
Ridge linear_model ✅ ❌
RidgeCV linear_model ✅ ❌
RidgeClassifier linear_model ✅ ❌
RidgeClassifierCV linear_model ✅ ❌
RobustScaler preprocessing ✅ ❌
SGDClassifier linear_model ✅ ❌
SGDOneClassSVM linear_model ✅ ❌
SGDRegressor linear_model ✅ ❌
SVC svm ✅ ❌
SVR svm ✅ ❌
SelectFdr feature_selection ✅ ❌
SelectFpr feature_selection ✅ ❌
SelectFromModel feature_selection ✅ ❌
SelectFwe feature_selection ✅ ❌
SelectKBest feature_selection ✅ ❌
SelectPercentile feature_selection ✅ ❌
SelfTrainingClassifier semi_supervised ❌ ❌
SequentialFeatureSelector feature_selection ❌ ❌
ShrunkCovariance covariance ❌ ❌
SimpleImputer impute ✅ ❌
SkewedChi2Sampler kernel_approximation ❌ ❌
SparseCoder decomposition ❌ ❌
SparsePCA decomposition ❌ ❌
SparseRandomProjection random_projection ❌ ❌
SpectralBiclustering cluster ❌ ❌
SpectralClustering cluster ❌ ❌
SpectralCoclustering cluster ❌ ❌
SplineTransformer preprocessing ❌ ❌
StackingClassifier ensemble ✅ ❌
StackingRegressor ensemble ✅ ❌
StandardScaler preprocessing ✅ ❌
TargetEncoder preprocessing ❌ ❌
TheilSenRegressor linear_model ✅ ❌
TransformedTargetRegressor compose ❌ ❌
TruncatedSVD decomposition ✅ ❌
TunedThresholdClassifierCV model_selection ✅ ❌
TweedieRegressor linear_model ✅ ❌
VarianceThreshold feature_selection ✅ ❌
VotingClassifier ensemble ✅ ❌
VotingRegressor ensemble ✅ ❌
Number Of Models 194 131 0

About

Lightweight deployment and scalability of scikit-learn ONNX models using Go.

Resources

Stars

1 star

Watchers

0 watching

Forks

Releases

Packages

Used by

Contributors

Languages