python
/
nyaggle
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

replace removed function

feature/drop-old-python
Taiga Noumi 2023-07-10 23:33:06 +09:00
parent 902776c777
commit 428e69ef23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nyaggle/experiment/auto_prep.py
View File

@ -2,7 +2,7 @@ from typing import List, Optional, Tuple
import numpy as np
import pandas as pd
from pandas.api.types import is_integer_dtype, is_categorical
from pandas.api.types import is_integer_dtype, is_categorical_dtype
from sklearn.preprocessing import LabelEncoder
@ -24,7 +24,7 @@ def autoprep_gbdt(algorithm_type: str, X_train: pd.DataFrame, X_test: Optional[p
if algorithm_type == 'lgbm':
# LightGBM can handle categorical dtype natively
categorical_feature_to_treat = [c for c in categorical_feature_to_treat if not is_categorical(X_train[c])]
categorical_feature_to_treat = [c for c in categorical_feature_to_treat if not is_categorical_dtype(X_train[c])]
if algorithm_type == 'cat' and len(categorical_feature_to_treat) > 0:
X_train = X_train.copy()
@ -49,7 +49,7 @@ def autoprep_gbdt(algorithm_type: str, X_train: pd.DataFrame, X_test: Optional[p
def _fill_na_by_unique_value(strain: pd.Series, stest: Optional[pd.Series]) -> Tuple[pd.Series, pd.Series]:
if is_categorical(strain):
if is_categorical_dtype(strain):
return strain.cat.codes, stest.cat.codes
elif is_integer_dtype(strain.dtype):
fillval = min(strain.min(), stest.min()) - 1