Skip to content
Snippets Groups Projects
Unverified Commit 2a325003 authored by Dominique Benielli's avatar Dominique Benielli Committed by GitHub
Browse files

Update early_fusion_adaboost.py

parent 324236d8
No related branches found
No related tags found
No related merge requests found
......@@ -10,14 +10,14 @@ classifier_class_name = "EarlyFusionAdaboost"
class EarlyFusionAdaboost(BaseEarlyFusion):
def __init__(self, random_state=None, n_estimators=50,
base_estimator=None, base_estimator_config=None, **kwargs):
estimator=None, base_estimator_config=None, **kwargs):
BaseEarlyFusion.__init__(self, random_state=random_state,
monoview_classifier="adaboost",
n_estimators= n_estimators,
base_estimator=base_estimator,
estimator=estimator,
base_estimator_config=base_estimator_config, **kwargs)
self.param_names = ["n_estimators", "base_estimator"]
self.classed_params = ["base_estimator"]
self.param_names = ["n_estimators", "estimator"]
self.classed_params = ["estimator"]
self.distribs = [CustomRandint(low=1, high=500),
base_boosting_estimators]
self.weird_strings = {"base_estimator": "class_name"}
\ No newline at end of file
self.weird_strings = {"estimator": "class_name"}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment