Skip to content

Commit 381996b

Browse files
Merge pull request #122 from scikit-learn-contrib/rpca_transform
Rpca transform
2 parents ffdf737 + 2d7119b commit 381996b

File tree

20 files changed

+924
-1913
lines changed

20 files changed

+924
-1913
lines changed

docs/api.rst

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,9 @@ Imputers
2323
imputations.imputers.ImputerOracle
2424
imputations.imputers.ImputerRegressor
2525
imputations.imputers.ImputerResiduals
26-
imputations.imputers.ImputerRPCA
26+
imputations.imputers.ImputerRpcaPcp
27+
imputations.imputers.ImputerRpcaNoisy
28+
imputations.imputers.ImputerSoftImpute
2729
imputations.imputers.ImputerShuffle
2830

2931
Comparator

docs/imputers.rst

Lines changed: 40 additions & 28 deletions
Large diffs are not rendered by default.

examples/benchmark.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -127,13 +127,13 @@ imputer_spline = imputers.ImputerInterpolation(groups=("station",), method="spli
127127
imputer_shuffle = imputers.ImputerShuffle(groups=("station",))
128128
imputer_residuals = imputers.ImputerResiduals(groups=("station",), period=365, model_tsa="additive", extrapolate_trend="freq", method_interpolation="linear")
129129

130-
imputer_rpca = imputers.ImputerRPCA(groups=("station",), columnwise=False, max_iterations=500, tau=2, lam=0.05)
131-
imputer_rpca_opti = imputers.ImputerRPCA(groups=("station",), columnwise=False, max_iterations=256)
130+
imputer_rpca = imputers.ImputerRpcaNoisy(groups=("station",), columnwise=False, max_iterations=500, tau=2, lam=0.05)
131+
imputer_rpca_opti = imputers.ImputerRpcaNoisy(groups=("station",), columnwise=False, max_iterations=256)
132132
dict_config_opti["RPCA_opti"] = {
133133
"tau": ho.hp.uniform("tau", low=.5, high=5),
134134
"lam": ho.hp.uniform("lam", low=.1, high=1),
135135
}
136-
imputer_rpca_opticw = imputers.ImputerRPCA(groups=("station",), columnwise=False, max_iterations=256)
136+
imputer_rpca_opticw = imputers.ImputerRpcaNoisy(groups=("station",), columnwise=False, max_iterations=256)
137137
dict_config_opti["RPCA_opticw"] = {
138138
"tau/TEMP": ho.hp.uniform("tau/TEMP", low=.5, high=5),
139139
"tau/PRES": ho.hp.uniform("tau/PRES", low=.5, high=5),

0 commit comments

Comments
 (0)