Commit 5b49fe8d authored by Duc Cao's avatar Duc Cao

Compute only the time that takes to run the extraction

parent 35ed780d
......@@ -23,7 +23,7 @@ import fr.limsi.sourceExtractor.training.TrainingUnLabel;
*/
public class MainSourceExtractor {
// Default configuration file
private static final String DEFAULT_CONFIG_FILE = "config.properties";
private static final String DEFAULT_CONFIG_FILE = "src/main/resources/config.properties";
// Configuration fields
private static final String DATA_DIR_PROPERTY = "DATA_DIR";
......@@ -43,7 +43,6 @@ public class MainSourceExtractor {
}
public static void main(String[] args) throws IOException, MaltChainedException, InterruptedException, URISyntaxException {
long startTime = System.currentTimeMillis();
// Debug mode
if (args.length == 0) {
// String argsStr = "-u -b -c
......@@ -104,9 +103,15 @@ public class MainSourceExtractor {
config.loadTools();
config.loadWapitiNativeLib();
long startTime = System.currentTimeMillis();
if (cmd.hasOption(CLIParameters.OPTION_DATA_SPLIT)) {
TrainingLabel trainingLabel = new TrainingLabel(config);
trainingLabel.splitTrainDevTest();
}
// if labeled
if (cmd.hasOption(CLIParameters.OPTION_DATA_TYPE_LABELED)) {
else if (cmd.hasOption(CLIParameters.OPTION_DATA_TYPE_LABELED)) {
TrainingLabel trainingLabel = new TrainingLabel(config);
trainingLabel.train(modelSuffix, jubNumber);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment