U 3dY@sdddZdddZdS)NcCsv|dkrDt|dr*t|jdr*|j}qVddt|jdD}nt|drV|}tt|t|krrtd|S)a(Check feature names. Parameters ---------- X : array-like of shape (n_samples, n_features) Input data. feature_names : None or array-like of shape (n_names,), dtype=str Feature names to check or `None`. Returns ------- feature_names : list of str Feature names validated. If `feature_names` is `None`, then a list of feature names is provided, i.e. the column names of a pandas dataframe or a generic list of feature names (e.g. `["x0", "x1", ...]`) for a NumPy array. NcolumnstolistcSsg|]}d|qS)x).0irr@/tmp/pip-unpacked-wheel-zrfo1fqw/sklearn/inspection/_pd_utils.py sz(_check_feature_names..z,feature_names should not contain duplicates.)hasattrrrrangeshapelenset ValueError)X feature_namesrrr_check_feature_namess  rc Csht|trd|dkr"td|dz ||WStk rb}ztd|d|W5d}~XYnX|S)aGet feature index. Parameters ---------- fx : int or str Feature index or name. feature_names : list of str, default=None All feature names from which to search the indices. Returns ------- idx : int Feature index. Nz+Cannot plot partial dependence for feature z since the list of feature names was not provided, neither as column names of a pandas data-frame nor via the feature_names parameter.zFeature z not in feature_names) isinstancestrrindex)Zfxrerrr_get_feature_index$s   $r)N)N)rrrrrrs #