Commit 91cffb92 authored by Victor Zimmermann's avatar Victor Zimmermann
Browse files

Update absinth.py

parent 5e47d72d

src/absinth.py

+4 −3
Original line number Diff line number Diff line
@@ -949,7 +949,7 @@ def characteristic_path_length(graph: nx.Graph) -> float: 
    return np.mean(path_length_list)





def print_stats(stat_dict: dict) -> None:

def print_stats(stat_dict: dict, graph: nx.Graph) -> None:

    """Prints various statistics and logs them to file.

    

    Args:
@@ -970,11 +970,12 @@ def print_stats(stat_dict: dict) -> None: 
    stat_string.append('Global clustering coefficient: {}.'.format(stat_dict['C']))

    stat_string.append('Mean cluster length (arithmetic): {}.'.format(stat_dict['a_mean_size']))

    stat_string.append('Mean cluster length (harmonic): {}.'.format(stat_dict['h_mean_size']))

    stat_string.append('Mean node degree: {}.'.format(np.mean(list(graph.degree.values()))))

    stat_string.append('Number of clusters: {}.'.format(stat_dict['clusters']))

    stat_string.append('Tuples gained through merging: {}.'.format(stat_dict['pipe_gain']))

    stat_string.append('Sense inventory:')

    for hub in stat_dict['hubs'].keys():

        stat_string.append(' -> {}: {}.'.format(hub, ", ".join(stat_dict['hubs'][hub])))

        stat_string.append(' -> {} ({}): {}.'.format(hub,graph.degree[hub], ", ".join(stat_dict['hubs'][hub])))

    

    print('\n[A] '+'\n[A] '.join(stat_string)+'\n')
@@ -1122,7 +1123,7 @@ def main(topic_id: int, topic_name: str, result_dict: dict) -> None: 
                                                        topic_id, result_id)

                    output_file.write(output_line)

                    

        print_stats(stat_dict)

        print_stats(stat_dict, graph)