calculates_results_stats.py

def calculates_results_stats(results_dic):
    results_stats_dic = dict()

    results_stats_dic['n_dogs_img'] = 0
    results_stats_dic['n_match'] = 0
    results_stats_dic['n_correct_dogs'] = 0
    results_stats_dic['n_correct_notdogs'] = 0
    results_stats_dic['n_correct_breed'] = 0

    for key in results_dic:
        if results_dic[key][2] == 1:
            results_stats_dic['n_match'] += 1

        if sum(results_dic[key][2:]) == 3:
            results_stats_dic["n_correct_breed"] += 1

        if results_dic[key][3] == 1:
            results_stats_dic['n_dogs_img'] += 1
            if results_dic[key][4] == 1:
                results_stats_dic['n_correct_dogs'] += 1
        else:
            if results_dic[key][4] == 0:
                results_stats_dic["n_correct_notdogs"] += 1

    results_stats_dic['n_images'] = len(results_dic)
    results_stats_dic['n_notdogs_img'] = (results_stats_dic['n_images'] -
                                          results_stats_dic['n_dogs_img'])

    # Calculates % correct for matches
    results_stats_dic['pct_match'] = (results_stats_dic['n_match'] / results_stats_dic['n_images']) * 100.0

    # Calculates % correct dogs
    results_stats_dic['pct_correct_dogs'] = (results_stats_dic['n_correct_dogs'] / results_stats_dic['n_dogs_img']) * 100.0

    # Calculates % correct breed of dog
    results_stats_dic['pct_correct_breed']=(results_stats_dic['n_correct_breed'] / results_stats_dic['n_dogs_img'])*100.0


    # Calculates % correct not-a-dog images
    if results_stats_dic['n_notdogs_img'] > 0:
        results_stats_dic['pct_correct_notdogs'] = (results_stats_dic['n_correct_notdogs'] /
                                                    results_stats_dic['n_notdogs_img']) * 100.0
    else:
        results_stats_dic['pct_correct_notdogs'] = 0.0

    return results_stats_dic