escrito-docker/local/models/RunModelTraining.java
2023-10-07 15:11:50 +02:00

26 lines
784 B
Java

package de.unidue.ltl.escrito.examples.local.models;
public class RunModelTraining {
public static void main(String[] args) {
String pathToTrainConfigCSVFile = "/home/felix/Documents/work/hiwi/iwm-tuebingen/escrito-stuff/model_training/config_files/config.csv";
String pathToDatasets = "/home/felix/Documents/work/hiwi/iwm-tuebingen/escrito-stuff/dkpro_target/datasets/N=704_10Ref_ChatGPT_augmented";
TrainConfig[] trainConfigs = TrainConfig.createTrainConfigsFromCSV(pathToTrainConfigCSVFile);
/*
for (TrainConfig trainCfg : trainConfigs) {
System.out.println(trainCfg);
}
*/
TrainAndSaveModel modelTrainer = new TrainAndSaveModel(pathToDatasets);
for (TrainConfig trainCfg : trainConfigs) {
modelTrainer.runModelTraining(trainCfg);
}
}
}