Skip to content
This repository was archived by the owner on Aug 25, 2024. It is now read-only.

Commit c173a9f

Browse files
committed
model: scikit: Expose RandomForestRegressor as scikitrfr
Signed-off-by: John Andersen <[email protected]>
1 parent 541a70a commit c173a9f

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

model/scikit/dffml_model_scikit/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,8 @@
5555
| +-------------------------------+----------------+-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------+
5656
| | ElasticNet | scikiteln | `scikiteln <https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.ElasticNet.html#sklearn.linear_model.ElasticNet/>`_ |
5757
| +-------------------------------+----------------+-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------+
58+
| | RandomForestRegressor | scikitrfr | `scikitrfr <https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestRegressor.html>`_ |
59+
| +-------------------------------+----------------+-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------+
5860
| | BayesianRidge | scikitbyr | `scikitbyr <https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.BayesianRidge.html#sklearn.linear_model.BayesianRidge/>`_ |
5961
| +-------------------------------+----------------+-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------+
6062
| | Lasso | scikitlas | `scikitlas <https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.Lasso.html#sklearn.linear_model.Lasso/>`_ |

model/scikit/dffml_model_scikit/scikit_models.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
from sklearn.tree import DecisionTreeClassifier, DecisionTreeRegressor
1818
from sklearn.ensemble import (
1919
RandomForestClassifier,
20+
RandomForestRegressor,
2021
AdaBoostClassifier,
2122
GradientBoostingClassifier,
2223
ExtraTreesClassifier,
@@ -72,6 +73,7 @@
7273
("scikitgpc", "GaussianProcessClassifier", GaussianProcessClassifier,),
7374
("scikitdtc", "DecisionTreeClassifier", DecisionTreeClassifier,),
7475
("scikitrfc", "RandomForestClassifier", RandomForestClassifier,),
76+
("scikitrfr", "RandomForestRegressor", RandomForestRegressor,),
7577
("scikitmlp", "MLPClassifier", MLPClassifier),
7678
("scikitgnb", "GaussianNB", GaussianNB),
7779
(

0 commit comments

Comments
 (0)