riccotti/LORE

recognize_features_type in util.py must recognize int32 and int64

rodrigocostacamargos opened this issue · 0 comments

def recognize_features_type(df, class_name):
int_types = ['int64', 'int32'] # rodrigo
integer_features = list(df.select_dtypes(include=int_types).columns)
# integer_features = list(df.select_dtypes(include=['int64']).columns)
double_features = list(df.select_dtypes(include=['float64']).columns)
string_features = list(df.select_dtypes(include=['object']).columns)
type_features = {
'integer': integer_features,
'double': double_features,
'string': string_features,
}