gen / scripts /train_pert.py
jaothan's picture
Upload 24 files
6f20934 verified
from data.datasets import load_dataset, preprocess_datafrom models.pert_model import train_pert_model# Load and preprocess datatrain_data = preprocess_data(load_dataset('data/train.csv'))val_data = preprocess_data(load_dataset('data/val.csv'))# Train the modelmodel = train_pert_model(train_data, val_data)