Skip to content

additional unit test for NeurIPS benchmarking paper #26

Open
@janvanrijn

Description

@janvanrijn

TODO: add unit test

package openmlweka;

import org.apache.commons.lang3.tuple.Pair;
import org.junit.Test;
import org.openml.apiconnector.io.OpenmlConnector;
import org.openml.apiconnector.xml.Run;
import org.openml.apiconnector.xml.Study;
import org.openml.apiconnector.xml.Task;
import org.openml.weka.algorithm.InstancesHelper;
import org.openml.weka.algorithm.WekaConfig;
import org.openml.weka.experiment.RunOpenmlJob;

import weka.classifiers.Classifier;
import weka.classifiers.trees.REPTree;
import weka.core.Instances;


public class TestOpenMLBenchmarkingSuites extends BaseTestFramework {

	
	@Test
	public void NeurIPSBenchmarkPaperCodeSample() throws Exception {
		OpenmlConnector openml = client_write_test; // OpenmlConnector openml = new OpenMLConnector();
		  Study benchmarksuite = openml.studyGet("OpenML100", "tasks"); // OpenML-CC18          // obtain the benchmark suite
		  Classifier tree = new REPTree();                                          // build a Weka classifier
		  for (Integer taskId : benchmarksuite.getTasks()) {                        // iterate over all tasks
		    Task t = openml.taskGet(taskId);                                        // download the OpenML task
		    if (t.getTask_id() > 10) { break; }
		    Instances d = InstancesHelper.getDatasetFromTask(openml, t);            // obtain the dataset
		    // openml.setApiKey("FILL_IN_OPENML_API_KEY");
		    Pair<Integer, Run> result = RunOpenmlJob.executeTask(openml, new WekaConfig("skip_jvm_benchmark=true; avoid_duplicate_runs=false"), taskId, tree);
		    Run run = openml.runGet(result.getLeft());}                             // retrieve the uploaded run
	}
} 

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions