/analysis-tools

Analysis tools for machine learning projects

Primary LanguagePython

Analysis tools for machine learning projects

1. Usage

$ pip install analysis-tools

2. Tutorial

examples/titanic/eda.ipynb를 참고

from analysis_tools import eda, metrics

data   = pd.DataFrame(..)
target = 'survived'

num_features       = ['age', 'sibsp', 'parch', 'fare']
cat_features       = data.columns.drop(num_features)
data[num_features] = data[num_features].astype('float32')
data[cat_features] = data[cat_features].astype('string')

eda.plot_missing_value(data)
eda.plot_features(data)
eda.plot_features_target(data, target)
eda.plot_corr(data.corr())
metrics.get_feature_importance(data, target)