The Wayback Machine - https://web.archive.org/web/20210105073127/https://github.com/openml/openml-pytorch
Skip to content
master
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 

README.md

Pytorch extension for OpenML python

Pytorch extension for openml-python API.

Installation Instructions:

pip install openml-pytorch

PyPi link https://pypi.org/project/openml-pytorch/

Usage

Import openML libraries

import openml
import openml_pytorch
import openml_pytorch.layers
import openml_pytorch.config

Create a torch model

model = torch.nn.Sequential(
    processing_net,
    features_net,
    results_net
)

Download the task from openML and run the model on task.

task = openml.tasks.get_task(3573)
run = openml.runs.run_model_on_task(model, task, avoid_duplicate_runs=False)
run.publish()
print('URL for run: %s/run/%d' % (openml.config.server, run.run_id))

Note: The input layer of the network should be compatible with OpenML data output shape. Please check examples for more information.

About

Pytorch extension for openml-python

Topics

Resources

License

Releases

No releases published

Packages

No packages published

Languages

You can’t perform that action at this time.