A collection of regression datasets
Project description
RegData
A collection of regression datasets.
Install
pip install regdata
Quick example
import regdata as rd
rd.set_backend('torch') # numpy, tf (numpy is default)
X, y, X_test = rd.Step().get_data() # Loads step function dataset
Features
- Simple API for quick benchmarking on various datasets.
- Get data in any framework:
torch
,tensorflow
ornumpy
by setting a global backend. - Scale
X
and/ory
data withMinMaxScaler
orStandardScaler
. - Get
y
in squeezed(n,)
or unsqueezed(n,1)
format. - Perform only mean normalization on
y
.
Plot datasets to have a quick glance
import regdata as rd
rd.Olympic().plot()
Checkout all plots here.
Datasets
from regdata import (
DellaGattaGene,
MotorcycleHelmet,
Olympic,
SineJump1D,
Smooth1D,
Step
)
References
- DellaGattaGene
- MotorcycleHelmet
- Olympic
- SineJump1D
- Smooth1D - Example 2
- Step
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
regdata-1.0.0.tar.gz
(88.8 kB
view hashes)