Skip to content

Commit ed3e55e

Browse files
Merge pull request #73 from ymouad/master
docs: Update _kliep.py
2 parents 44c9b67 + 079eefa commit ed3e55e

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

adapt/instance_based/_kliep.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,14 @@ class KLIEP(BaseAdaptEstimator):
9898
Optimization algorithm.
9999
Possible values: ['original', 'PG', 'FW']
100100
101+
- 'original' follows the algorithm of [1]. Useful to reproduce the paper's experiences.
102+
- 'PG' is a improved version of 'original'. A convex projection into the constraints set is used.
103+
- 'FW' [2] uses the Frank-Wolfe algorithm to solve the above OP.
104+
105+
In general, 'FW' is more efficient than 'original' or 'PG'.
106+
In some cases, 'PG' converges faster than 'FW' with a good choice of learning rate.
107+
108+
101109
lr : float or list of float (default=np.logspace(-3,1,5))
102110
Learning rate of the gradient ascent.
103111
Used only if algo different to 'FW'
@@ -201,7 +209,10 @@ class KLIEP(BaseAdaptEstimator):
201209
M. Sugiyama, S. Nakajima, H. Kashima, P. von Bünau and M. Kawanabe. \
202210
"Direct importance estimation with model selection and its application \
203211
to covariateshift adaptation". In NIPS 2007
204-
"""
212+
.. [2] `[2] <https://webdocs.cs.ualberta.ca/~dale/papers/ijcai15.pdf>`_ \
213+
J. Wen, R. Greiner and D. Schuurmans. \
214+
"Correcting Covariate Shift with the Frank-Wolfe Algorithm". In IJCAI 2015
215+
"""
205216
def __init__(self,
206217
estimator=None,
207218
Xt=None,

0 commit comments

Comments
 (0)