Added missing declaration for decision_tree_forest
#1
by
something456
- opened
README.md
CHANGED
@@ -39,7 +39,8 @@ config = AutoConfig.from_pretrained(model_name_or_path)
|
|
39 |
model = MetaTree.from_pretrained(
|
40 |
model_name_or_path,
|
41 |
config=config,
|
42 |
-
)
|
|
|
43 |
|
44 |
# Load Datasets
|
45 |
X, y, feature_names = imodels.get_clean_dataset('fico', data_source='imodels')
|
|
|
39 |
model = MetaTree.from_pretrained(
|
40 |
model_name_or_path,
|
41 |
config=config,
|
42 |
+
)
|
43 |
+
decision_tree_forest = DecisionTreeForest()
|
44 |
|
45 |
# Load Datasets
|
46 |
X, y, feature_names = imodels.get_clean_dataset('fico', data_source='imodels')
|