init_model_stats

init_model_stats.py

def init_model_stats(id2label):
    return {
        label: {
            'total': 0,
            'true_positives': 0,
            'false_positives': 0,
            'false_negatives': 0,
            'accuracy': 0.0,
            'f1': 0.0
        }
        for label_id, label in id2label.items()
    }