julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update Iris dataset

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-05-17 16:57:13 +02:00
parent 768d969f89
commit 639198e774
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
prototorch/datasets/iris.py
View File

@ -10,6 +10,8 @@ from sklearn.datasets import load_iris
class Iris(NumpyDataset): class Iris(NumpyDataset):
def __init__(self): def __init__(self, dims=None):
x, y = load_iris(return_X_y=True) x, y = load_iris(return_X_y=True)
if dims:
x = x[:, dims]
super().__init__(x, y) super().__init__(x, y)