PythonGenericRegressor.java 文件源码

/**
* 
* @param filename : the conifiguration file name for required to run xgboost from the command line
* @param datset : the dataset to be used
* @param model : model dump name
* @param columns : Number of columns in the data
*/
 private void create_config_file(String filename , String datset, String model, int columns){

try{  // Catch errors in I/O if necessary.
      // Open a file to write to.

        String saveFile = filename;

        FileWriter writer = new FileWriter(saveFile);
        writer.append("task=train\n");
        writer.append("columns=" + columns+ "\n");                  
        writer.append("model=" +model+ "\n");   
        writer.append("data=" +datset+ "\n");                   
        writer.append("n_jobs=" + this.threads + "\n");
        writer.append("random_state=" +  this.seed + "\n");
        if (this.verbose){
            writer.append("verbose=1" +  "\n");
        }else {
            writer.append("verbose=0" +  "\n");                 
        }           
        writer.close();

    } catch (Exception e) {
        throw new IllegalStateException(" failed to write the config file at: " + filename);
    }    
 }