Browse Source

Update sklearn.js

Lutz Roeder 6 years ago
parent
commit
fe0903ccb2
1 changed files with 10 additions and 2 deletions
  1. 10 2
      src/sklearn.js

+ 10 - 2
src/sklearn.js

@@ -173,7 +173,12 @@ sklearn.ModelFactory = class {
                 constructorTable['sklearn.discriminant_analysis.LinearDiscriminantAnalysis'] = function() {};
                 constructorTable['sklearn.externals.joblib.numpy_pickle.NumpyArrayWrapper'] = constructorTable['joblib.numpy_pickle.NumpyArrayWrapper'];
                 constructorTable['sklearn.externals.joblib.numpy_pickle.NDArrayWrapper'] = function() {};
+                constructorTable['sklearn.ensemble._bagging.BaggingClassifier'] = function() {};
                 constructorTable['sklearn.ensemble._forest.RandomForestClassifier'] = function() {};
+                constructorTable['sklearn.ensemble._forest.ExtraTreesClassifier'] = function() {};
+                constructorTable['sklearn.ensemble._gb_losses.BinomialDeviance'] = function() {};
+                constructorTable['sklearn.ensemble._gb.GradientBoostingClassifier'] = function() {};
+                constructorTable['sklearn.ensemble._voting.VotingClassifier'] = function() {};
                 constructorTable['sklearn.ensemble.forest.RandomForestClassifier'] = function() {};
                 constructorTable['sklearn.ensemble.forest.RandomForestRegressor'] = function() {};
                 constructorTable['sklearn.ensemble.forest.ExtraTreesClassifier'] = function() {};
@@ -213,12 +218,13 @@ sklearn.ModelFactory = class {
                 constructorTable['sklearn.neighbors.kd_tree.newObj'] = function() {};
                 constructorTable['sklearn.neighbors.KNeighborsClassifier​'] = function() {};
                 constructorTable['sklearn.neighbors.KNeighborsRegressor'] = function() {};
+                constructorTable['sklearn.neural_network._multilayer_perceptron.MLPClassifier'] = function() {};
+                constructorTable['sklearn.neural_network._stochastic_optimizers.AdamOptimizer'] = function() {};
+                constructorTable['sklearn.neural_network._stochastic_optimizers.SGDOptimizer'] = function() {};
                 constructorTable['sklearn.neural_network.rbm.BernoulliRBM'] = function() {};
                 constructorTable['sklearn.neural_network.multilayer_perceptron.MLPClassifier'] = function() {};
                 constructorTable['sklearn.neural_network.multilayer_perceptron.MLPRegressor'] = function() {};
                 constructorTable['sklearn.neural_network.stochastic_gradient.SGDClassifier'] = function() {};
-                constructorTable['sklearn.neural_network._stochastic_optimizers.AdamOptimizer'] = function() {};
-                constructorTable['sklearn.neural_network._stochastic_optimizers.SGDOptimizer'] = function() {};
                 constructorTable['sklearn.pipeline.Pipeline'] = function() {};
                 constructorTable['sklearn.pipeline.FeatureUnion'] = function() {};
                 constructorTable['sklearn.preprocessing._data.RobustScaler'] = function() {};
@@ -246,6 +252,8 @@ sklearn.ModelFactory = class {
                 constructorTable['sklearn.svm.classes.SVC'] = function() {};
                 constructorTable['sklearn.svm.classes.SVR'] = function() {};
                 constructorTable['sklearn.tree._classes.DecisionTreeClassifier'] = function() {};
+                constructorTable['sklearn.tree._classes.DecisionTreeRegressor'] = function() {};
+                constructorTable['sklearn.tree._classes.ExtraTreeClassifier'] = function() {};
                 constructorTable['sklearn.tree._tree.Tree'] = function(n_features, n_classes, n_outputs) {
                     this.n_features = n_features;
                     this.n_classes = n_classes;