deepbox/ml
Ensembles
Boosting, bagging, stacking, and voting estimators for regression and classification.
Ensembles
AdaBoostClassifier
AdaBoost Classifier (Adaptive Boosting).
AdaBoostRegressor
AdaBoost Regressor (Adaptive Boosting for Regression).
BaggingClassifier
Bagging Classifier (Bootstrap Aggregating).
BaggingRegressor
Bagging Regressor (Bootstrap Aggregating for Regression).
GradientBoostingClassifier
Gradient Boosting Classifier.
GradientBoostingRegressor
Gradient Boosting Regressor.
StackingClassifier
Stacking Classifier.
StackingRegressor
Stacking Regressor.
VotingClassifier
Voting Classifier.
VotingRegressor
Voting Regressor.
ml-ensemble.ts
import { AdaBoostClassifier, BaggingClassifier, GradientBoostingClassifier, LogisticRegression, VotingClassifier,} from "deepbox/ml";import { tensor } from "deepbox/ndarray";const X = tensor([[1, 2], [2, 1], [3, 3], [4, 4]]);const y = tensor([0, 0, 1, 1]);console.log(new AdaBoostClassifier().fit(X, y).predict(X).toString());console.log(new BaggingClassifier().fit(X, y).predict(X).toString());console.log(new GradientBoostingClassifier().fit(X, y).predict(X).toString());console.log( new VotingClassifier({ estimators: [new LogisticRegression(), new AdaBoostClassifier()], }) .fit(X, y) .predict(X) .toString());