This repository has been archived on 2024-10-22. You can view files and clone it, but cannot push or open issues or pull requests.
soft-analytics-02/plot_acc.py
Claudio Maggioni a4ceee8716 Final version of the project
History has been rewritten to delete large files in repo
2024-01-03 15:28:43 +01:00

16 lines
415 B
Python

import os.path
import pandas as pd
from train.finetune import plot_loss_acc
ROOT = os.path.dirname(__file__)
def main():
df = pd.read_csv(os.path.join(ROOT, 'models', 'final', 'stats.csv'))
plot_loss_acc(df['train_loss'].tolist(), df['val_loss'].tolist(), df['train_acc'].tolist(), df['val_acc'].tolist(),
os.path.join(ROOT, 'models', 'final'))
if __name__ == "__main__":
main()