projects
/
pytorch.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Added the 5 cluster data-set.
[pytorch.git]
/
denoising-ae-field.py
diff --git
a/denoising-ae-field.py
b/denoising-ae-field.py
index
cc4af81
..
effee19
100755
(executable)
--- a/
denoising-ae-field.py
+++ b/
denoising-ae-field.py
@@
-31,10
+31,19
@@
def data_spiral(nb):
data = torch.cat((y, x), 1)
return data
data = torch.cat((y, x), 1)
return data
+def data_penta(nb):
+ a = (torch.randint(5, (nb,)).float() / 5 * 2 * math.pi).view(-1, 1)
+ x = a.cos()
+ y = a.sin()
+ data = torch.cat((y, x), 1)
+ data = data + data.new(data.size()).normal_(0, 0.05)
+ return data
+
######################################################################
######################################################################
-# data = data_spiral(1000)
-data = data_zigzag(1000)
+data = data_spiral(1000)
+# data = data_zigzag(1000)
+# data = data_penta(1000)
data = data - data.mean(0)
data = data - data.mean(0)