GitLab

Created by: MAGLeb

def test_unshaffled_data():
    target_column = 'species'
    df_el, y = load_iris(return_X_y=True, as_frame=True)
    df_el[target_column] = LabelEncoder().fit_transform(y)

    features, target = df_el.drop(target_column, axis=1).values, df_el[target_column].values

    problem = 'classification'
    auto_model = Fedot(problem=problem, seed=42, timeout=0.005)
    pipeline = auto_model.fit(features=features, target=target)
    assert pipeline is not None

work 1 minute or more

def test_unshaffled_data():
    target_column = 'species'
    df_el, y = load_iris(return_X_y=True, as_frame=True)
    df_el[target_column] = LabelEncoder().fit_transform(y)

    features, target = df_el.drop(target_column, axis=1).values, df_el[target_column].values

    problem = 'classification'
    auto_model = Fedot(problem=problem, seed=42, composer_params={'timeout': 0.005})
    pipeline = auto_model.fit(features=features, target=target)
    assert pipeline is not None

work 5 miliseconds