org.dkpro.tc.ml.xgboost.core

Class XgboostTrainer

java.lang.Object

org.dkpro.tc.ml.xgboost.core.Xgboost

org.dkpro.tc.ml.xgboost.core.XgboostTrainer

All Implemented Interfaces:

TcTrainer

public class XgboostTrainer
extends Xgboost
implements TcTrainer

Constructor Summary

Constructors

Constructor and Description
XgboostTrainer()

Method Summary

Modifier and Type	Method and Description
void	train(File data, File model, List<String> parameters) Trains a model with Xgboost
File	writeTrainConfigurationFile(List<String> parameters, File data, File model)

Methods inherited from class org.dkpro.tc.ml.xgboost.core.Xgboost

flipBackslash, getExecutable, runCommand, uninstallExecutable

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

XgboostTrainer

public XgboostTrainer()

Method Detail

train

public void train(File data,
                  File model,
                  List<String> parameters)
           throws Exception

Trains a model with Xgboost

Specified by:

train in interface TcTrainer

Parameters:

data - The training data file

model - File descriptor for the location at which the model shall be stored

parameters - The parametrization

Throws:

Exception - In case of an error

writeTrainConfigurationFile

public File writeTrainConfigurationFile(List<String> parameters,
                                        File data,
                                        File model)
                                 throws Exception

Throws:

Exception