We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ed69a7a commit 7d8db2cCopy full SHA for 7d8db2c
1 file changed
dice_ml/data_interfaces/private_data_interface.py
@@ -52,7 +52,7 @@ def __init__(self, params):
52
self.categorical_levels = {}
53
54
for feature in features_dict:
55
- if isinstance(features_dict[feature][0], int): # continuous feature
+ if isinstance(features_dict[feature][0], (int, float)): # continuous feature
56
self.continuous_feature_names.append(feature)
57
else:
58
self.categorical_feature_names.append(feature)
0 commit comments