개발/AI

[AutoViz] EDA를 도와줄 시각화 툴

jykim23 2023. 10. 31. 19:09
728x90
!pip install AutoViz
!pip install xlrd

import warnings
warnings.filterwarnings('ignore')
import pandas as pd
from autoviz.AutoViz_Class import AutoViz_Class
AV = AutoViz_Class()

# 데이터 불러오기
file = "csv 파일"
df = pd.read_pickle(file)

df.head()

# AutoViz 실행
save_viz_dir = 'eda_viz' # 폴더 이름. chart_format 형식으로 저장될 파일 위치.
dftc = AV.AutoViz(filename='', 
                  sep=',' , 
                  depVar='Attrition_Flag', # 타겟 데이터 컬럼명
                  dfte=df, # 데이터
                  header=0, 
                  verbose=2, # 정보량(?)
                  lowess=False, 
                  chart_format='html', # png 등 가능
                  max_rows_analyzed=15000, 
                  max_cols_analyzed=30,
                  save_plot_dir=save_viz_dir
)

 

 

 

728x90