def train_automl()

in Project-AutoML/automl/mod/mod_autosklearn.py [0:0]

• 22 lines of code
• 3 McCabe index (conditional complexity)

    def train_automl(train_x, train_y, other_params=None, **kwargs):
        params = Params(param_str=other_params, **kwargs)
        print(params)
        pipeline_mods = []

        pipeline_mods.append(
            (
                "oridinal_encoder",
                OrdinalEncoder(
                    unknown_value=np.nan, handle_unknown="use_encoded_value"
                ),
            )
        )
        pipeline = Pipeline(steps=pipeline_mods)
        feat_type = [
            "Categorical" if x.name in {"object", "category"} else "Numerical"
            for x in train_x.dtypes
        ]
        train_x = pipeline.fit_transform(train_x)
        classifier = AutoSklearnClassifier(**params.input_params)
        classifier.fit(train_x, train_y, feat_type=feat_type)

        pipeline.steps.append(("classifier", classifier))
        return pipeline