def get_best_training_job()

in sagemaker/source/visualization/model_visualisation_utils.py [0:0]

• 21 lines of code
• 7 McCabe index (conditional complexity)

def get_best_training_job(df_list, metric, maximize_or_minimize):
    '''
    Helper function to get the best training job.
    '''
    assert maximize_or_minimize in ["maximize", "minimize"], "maximize_or_minimize must be either 'maximize' or 'minimize'"
    if maximize_or_minimize == "maximize":
        best_value = 0
    else:
        best_value = 1e5
        
    best_job = None
        
    for job_name, job_df in df_list:
        if metric not in job_df.columns:
            continue
        final_value = job_df[metric].values[-1]
        if maximize_or_minimize == "maximize":
            if final_value > best_value:
                best_job = job_name, job_df
        else:
            if final_value < best_value:
                best_job = job_name, job_df
    return best_job