o 2hZ @sddlZzddlZddlmZddlmZWney/dZGdddZGdddZYnwdd Zd d ZGd d d eeZ dS)N) BaseEstimator)TransformerMixinc@ eZdZdS)rN__name__ __module__ __qualname__r r S/var/www/html/chatgem/venv/lib/python3.10/site-packages/keras/src/wrappers/utils.pyr rc@r)rNrr r r r r r rcCstdur t|ddS)NzW requires `scikit-learn` to be installed. Run `pip install scikit-learn` to install it.)sklearn ImportError) symbol_namer r r assert_sklearn_installeds rcCs|jr |jr |js tddS)z-Check whether the model need sto be compiled.zCGiven model needs to be compiled, and have a loss and an optimizer.N)compiledloss optimizer RuntimeError)modelr r r _check_models rc@s(eZdZdZddZddZddZdS) TargetReshaperaConvert 1D targets to 2D and back. For use in pipelines with transformers that only accept 2D inputs, like OneHotEncoder and OrdinalEncoder. Attributes: ndim_ : int Dimensions of y that the transformer was trained on. cCs |j|_|S)zFit the transformer to a target y. Returns: TargetReshaper A reference to the current instance of TargetReshaper. )ndimndim_selfyr r r fit.szTargetReshaper.fitcCs|jdkr |ddS|S)zMakes 1D y 2D. Args: y : np.ndarray Target y to be transformed. Returns: np.ndarray A numpy array, of dimension at least 2. )rreshaperr r r transform8s zTargetReshaper.transformcCs:ddlm}|||jdkr|jdkrtj|ddS|S)aRevert the transformation of transform. Args: y: np.ndarray Transformed numpy array. Returns: np.ndarray If the transformer was fit to a 1D numpy array, and a 2D numpy array with a singleton second dimension is passed, it will be squeezed back to 1D. Otherwise, it will eb left untouched. r)check_is_fittedr)axis)sklearn.utils.validationr!rrnpsqueeze)rrr!r r r inverse_transformGs z TargetReshaper.inverse_transformN)rrr__doc__rr r'r r r r r#s   r) numpyr%r sklearn.baserrr rrrr r r r s