a28601751e
* chore: Absolute imports * feat: Add new mesh util * chore: replace bumpversion original fork no longer maintained, move config * ci: remove old configuration files * ci: update github action * ci: add python 3.10 test * chore: update pre-commit hooks * ci: update supported python versions supported are 3.7, 3.8 and 3.9. 3.6 had EOL in december 2021. 3.10 has no pytorch distribution yet. * ci: add windows test * ci: update action less windows tests, pre commit * ci: fix typo * chore: run precommit for all files * ci: two step tests * ci: compatibility waits for style * fix: init file had missing imports * ci: add deployment script * ci: skip complete publish step * ci: cleanup readme
20 lines
535 B
Python
20 lines
535 B
Python
"""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)
|