File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -33,6 +33,7 @@ from joblib import Parallel, delayed
3333from packaging import version
3434from sklearn.model_selection import train_test_split
3535from sklearn.utils import check_X_y, check_array, check_random_state
36+ import numbers
3637
3738if version.parse(sklearn.__version__) >= version.parse(' 0.22.0' ):
3839 from sklearn.utils._testing import ignore_warnings
@@ -2264,7 +2265,7 @@ class UpliftTreeClassifier:
22642265 else :
22652266 v = observations[tree.col]
22662267 branch = None
2267- if isinstance (v, int ) or isinstance (v, float ):
2268+ if isinstance (v, numbers.Number ):
22682269 if v >= tree.value:
22692270 branch = tree.trueBranch
22702271 else :
@@ -2311,7 +2312,7 @@ class UpliftTreeClassifier:
23112312 return dict (result)
23122313 else :
23132314 branch = None
2314- if isinstance (v, int ) or isinstance (v, float ):
2315+ if isinstance (v, numbers.Number ):
23152316 if v >= tree.value:
23162317 branch = tree.trueBranch
23172318 else :
You can’t perform that action at this time.
0 commit comments