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)
