26 lines
784 B
Java
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);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|