Example #1
0
                }))
            stats.start_timing()
            # To avoid the output "first run crashed"...
            stats.submitted_ta_runs += 1
            stats.finished_ta_runs += 1
            memory_lim = memory_limit_factor * automl_arguments['memory_limit']

            pipeline, run_info, run_value = automl.fit_pipeline(
                X=X_train,
                y=y_train,
                X_test=X_test,
                y_test=y_test,
                resampling_strategy='test',
                memory_limit=memory_lim,
                disable_file_output=True,
                logger=logger,
                stats=stats,
                scoring_functions=scoring_functions,
                include=include,
                metric=automl_arguments['metric'],
                pynisher_context='spawn',
                cutoff=per_run_time_limit * 3,
                config=config,
            )

            if run_value.status == StatusType.SUCCESS:
                assert len(
                    run_value.additional_info) > 1, run_value.additional_info

            # print(additional_run_info)