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()