Skip to content

Commit 5dcc7e7

Browse files
author
[zebinyang]
committed
for a fair comparison, adjust roughness by map x to be within [0, 1]; version 0.1.8
1 parent 9f3a4df commit 5dcc7e7

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

simtree/__init__.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,12 @@
1+
from .cart import CARTRegressor, CARTClassifier
12
from .glmtree import GLMTreeRegressor, GLMTreeClassifier
3+
from .simtree import SIMTreeRegressor, SIMTreeClassifier
4+
from .customtree import CustomMobTreeRegressor, CustomMobTreeClassifier
25

3-
__all__ = ["GLMTreeRegressor", "GLMTreeClassifier"]
6+
__all__ = ["CARTRegressor", "CARTClassifier",
7+
"GLMTreeRegressor", "GLMTreeClassifier",
8+
"SIMTreeRegressor", "SIMTreeClassifier",
9+
"CustomMobTreeRegressor", "CustomMobTreeClassifier"]
410

5-
__version__ = '0.1'
11+
__version__ = '0.1.8'
612
__author__ = 'Zebin Yang'

0 commit comments

Comments
 (0)