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

feat: add XOR dataset

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-07-15 18:14:38 +02:00
parent c1d7cfee8f
commit bf09ff8f7f
No known key found for this signature in database
GPG Key ID: 3331B0F18B6D4D93
2 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
prototorch/datasets/__init__.py
View File

@ -10,3 +10,4 @@ from .sklearn import (
) )
from .spiral import Spiral from .spiral import Spiral
from .tecator import Tecator from .tecator import Tecator
from .xor import XOR

18
prototorch/datasets/xor.py Normal file
View File

@ -0,0 +1,18 @@
"""Exclusive-or (XOR) dataset for binary classification."""
import torch
def make_xor(num_samples=500):
x = torch.rand(num_samples, 2)
y = torch.zeros(num_samples)
y[torch.logical_and(x[:, 0] > 0.5, x[:, 1] < 0.5)] = 1
y[torch.logical_and(x[:, 1] > 0.5, x[:, 0] < 0.5)] = 1
return x, y
class XOR(torch.utils.data.TensorDataset):
"""Exclusive-or (XOR) dataset for binary classification."""
def __init__(self, num_samples: int = 500):
x, y = make_xor(num_samples)
super().__init__(x, y)