declare-lab/Multimodal-Infomax

acc/f1 calculation

cyZhu98 opened this issue · 1 comments

Hi, thank you for your great work! However, there seems to be a little mistake. accuracy_score, f1_score imported from sklearn.metrics should be:

f1_score / accuracy_score(y_true, y_pred)

which is from
https://github.com/declare-lab/Multimodal-Infomax/tree/main/src/utils/eval_metrics.py#L47

you can check it in https://scikit-learn.org/0.21/modules/classes.html#sklearn-metrics-metrics

Thanks for your pointing out. We have rerun the experiments and found f1_score is even higher. Therefore we preserve the previous results. We also have corrected the input.