We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c068605 commit 2771778Copy full SHA for 2771778
examples/local_pca_mnist.py
@@ -9,7 +9,7 @@
9
10
def get_subset_mnist(n: int = 1000):
11
dataset = MNIST(root="", download=True)
12
- data = dataset.data[dataset.targets==1]
+ data = dataset.data[dataset.targets == 1]
13
N = data.shape[0]
14
idx = np.random.choice(np.arange(N), size=n)
15
return data[idx]
0 commit comments