/mhddlZddlmZddlmZddlmZmZddl m Z ddl m Z ddl mZGdd eZGd d eZy) N)NDArray)Tuple)ABCabstractmethod)StandardScaler)Lasso) make_pipelinec <eZdZdZededededeeeffdZy) BaseSolverz A base solver class. Methods: fit(self, masks: NDArray, outputs: NDArray, num_output_tokens: int) -> Tuple[NDArray, NDArray]: Fit the solver to the given data. masksoutputsnum_output_tokensreturncyN)selfr r rs Tuple[NDArray, NDArray]: Fit the solver to the given data. lasso_alpharNc||_yr)r)rrs r__init__zLassoRegression.__init__&s &rr r rc|jtj}||z }t}t |j dd}t ||}|j|||j|jz } |j|j|jz |jjzz } | |z| |zfS)NrT)alpha random_state fit_intercept) astypenpfloat32rrrr rcoef_scale_ intercept_mean_T) rr r rXYscalerlassopipelineweightbiass rrzLassoRegression.fit)s LL $ ' '!D,,1DQ / Qv}},6<<&--#?5;;=="PP))42C+CCCr)g{Gz?) rrrrfloatr rrrrrrrrrsN 'E'T' D D'. DCF D w  Drr)numpyr& numpy.typingrtypingrabcrrsklearn.preprocessingrsklearn.linear_modelrsklearn.pipeliner r rrrrr<s4 #0&* & &DjDr