Closed pbarbarant closed 3 weeks ago
I'm unsure why this parameter is important ?
I'm unsure why this parameter is important ?
These are the lambdas in the variational formulation:
\min _\mu \sum_k \lambda_k d_{F G W, \alpha, 1, q}^{\mathbb{R}^d}\left(\mu, \mu_k\right)
In practice everyone gets assigned the same weight, it's essentially for the consistency with POT.
But if we're sure that we never change the weights, why not hard-coding the default (to unclutter the API) ?
In
FUGWBarycenter
, theweights_list
arguments is passed toupdate_barycenter_features
.weights_list
describes the probability weights in each subject and not the weight of the subject in the variational formulation of the barycenter. I propose to addsubject_weights
as an argument passed to thefit()
method in order to further avoid this confusion.