aimclub / FEDOT

Automated modeling and machine learning framework FEDOT
https://fedot.readthedocs.io
BSD 3-Clause "New" or "Revised" License
627 stars 86 forks source link

Timeout param in API does not work #425

Closed MAGLeb closed 2 years ago

MAGLeb commented 2 years ago
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

nicl-nno commented 2 years ago

Fixed