File tree Expand file tree Collapse file tree 1 file changed +12
-1
lines changed
Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments