def test_chain_fine_tune_all_nodes_correct(classification_dataset): data = classification_dataset first = PrimaryNode(operation_type='scaling') second = PrimaryNode(operation_type='knn') final = SecondaryNode(operation_type='dt', nodes_from=[first, second]) chain = Chain(final) iterations_total, time_limit_minutes = 5, 1 tuned_chain = chain.fine_tune_all_nodes(loss_function=roc, input_data=data, iterations=iterations_total, max_lead_time=time_limit_minutes) tuned_chain.predict(input_data=data) is_tuning_finished = True assert is_tuning_finished
def tune_chain(chain: Chain, data: InputData, time_limit): chain.fine_tune_all_nodes(input_data=data, max_lead_time=time_limit)