a | b/exp/nb_denoiser.py | ||
---|---|---|---|
1 | |||
2 | ################################################# |
||
3 | ### THIS FILE WAS AUTOGENERATED! DO NOT EDIT! ### |
||
4 | ################################################# |
||
5 | # file to edit: dev_nb/denoiser.ipynb |
||
6 | |||
7 | from sklearn.decomposition import TruncatedSVD |
||
8 | |||
9 | def denoise(data): |
||
10 | svd = TruncatedSVD(n_components=1, n_iter=7, random_state=0) |
||
11 | svd.fit(data) |
||
12 | pc = svd.components_ |
||
13 | data -= data.dot(pc.T) * pc |
||
14 | return data |