-
Notifications
You must be signed in to change notification settings - Fork 30
Description
File /opt/anaconda3/lib/python3.12/site-packages/SparseSC/fit_fast.py:131, in fit_fast(features, targets, model_type, treated_units, w_pens, custom_donor_pool, match_space_maker, w_pen_inner, avoid_NxN_mats, verbose, targets_aux, **kwargs)
129 def _fit_fast_wrapper(MatchSpace, V):
130 return _fit_fast_inner(X, MatchSpace.transform(X), Y, V, model_type, treated_units, w_pens=w_pens, custom_donor_pool=custom_donor_pool, w_pen_inner=w_pen_inner)
--> 131 MatchSpace, V, best_v_pen, MatchSpaceDesc = match_space_maker(X_v, Y_v, fit_model_wrapper=_fit_fast_wrapper, X_full=X, D_full=D)
132 match_fit = None
133 if isinstance(MatchSpaceDesc, tuple): #unpack if necessary
File /opt/anaconda3/lib/python3.12/site-packages/SparseSC/utils/match_space.py:135, in MTLassoCV_MatchSpace_factory.._MTLassoCV_MatchSpace_wrapper(X, Y, **kwargs)
134 def _MTLassoCV_MatchSpace_wrapper(X, Y, **kwargs):
--> 135 return _MTLassoCV_MatchSpace(
136 X, Y, v_pens=v_pens, n_v_cv=n_v_cv, sample_frac=sample_frac, Y_col_block_size=Y_col_block_size, se_factor=se_factor, normalize=normalize, fit_args=fit_args, **kwargs
137 )
File /opt/anaconda3/lib/python3.12/site-packages/SparseSC/utils/match_space.py:153, in _MTLassoCV_MatchSpace(X, Y, v_pens, n_v_cv, sample_frac, Y_col_block_size, se_factor, normalize, fit_args, **kwargs)
151 if Y_col_block_size is not None:
152 Y = block_summ_cols(Y, Y_col_block_size)
--> 153 varselectorfit = MultiTaskLassoCV(normalize=normalize, cv=n_v_cv, alphas=v_pens, **fit_args).fit(
154 X, Y
155 )
156 best_v_pen = varselectorfit.alpha
157 if se_factor is not None:
TypeError: MultiTaskLassoCV.init() got an unexpected keyword argument 'normalize'