Added missing declaration for decision_tree_forest (#1)
Browse files- Added missing declaration for decision_tree_forest (d8e632ef6dfc0d0fd285f7434021b34c45489eda)
Co-authored-by: Akshat Gupta <something456@users.noreply.huggingface.co>
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')
|