개발/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