Skip to content
Snippets Groups Projects
Commit d6d303ec authored by Luc Giffon's avatar Luc Giffon
Browse files

delete useless main

parent 8d0b3931
No related branches found
No related tags found
1 merge request!23Resolve "integration-sota"
......@@ -122,14 +122,3 @@ class KMeansForestClassifier(KmeansForest, metaclass=ABCMeta):
score = self._score_metric(y_preds, y_true)
best_tree_index = np.argmax(score) # get best scoring tree (the one with lowest mse)
return best_tree_index
\ No newline at end of file
if __name__ == "__main__":
from sklearn import datasets
from bolsonaro.models.model_parameters import ModelParameters
X, y = datasets.fetch_california_housing(return_X_y=True)
ModelParameters(extracted_forest_size=100,
normalize_D=True,
)
k_reg = KMeansForestRegressor()
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment