(size, noise=1.0)
| 2 | from sklearn.datasets import make_swiss_roll |
| 3 | |
| 4 | def sample_batch(size, noise=1.0): |
| 5 | x, _= make_swiss_roll(size, noise=noise) |
| 6 | return x[:, [0, 2]] / 10.0 |
| 7 | |
| 8 | def extract(input, t, x): |
| 9 | shape = x.shape |
nothing calls this directly
no outgoing calls
no test coverage detected