์ด์ ํฌ์คํธ ์์ฝ
์ง๋ ํฌ์คํธ์์ ์ฐ๋ฆฌ๋ A์ฌ์ ์ธ์ฌ ํจ์จํ๋ฅผ ์ํด ๊ธ์ฌ ์ฒด๊ณ ๊ด๋ จํ ๋ฌธ์ ๋ฅผ ์ฐพ์ ๋ด๊ณ ์ผ์ฐจ์ ์ธ ์คํ ํ๋ก์ธ์ค๋ฅผ ๊ตฌ์ถํ๋ค. ๊ทธ๋ฆฌ๊ณ ๊ฐ์ง๊ณ ์๋ ๋ฐ์ดํฐ๋ฅผ ์ ๋ชจ์์ ์ปดํจํฐ๊ฐ ์ดํดํ ์ ์๋ ๋ฐฉ์์ผ๋ก ๊ฐ๊ณตํด ์คฌ๋ค. null ๊ฐ, ์ค๋ณต ๊ฐ, outlier(์ ์ ์์น๋ฅผ ๋์ด์ ๋ฐ์ดํฐ)๊ฐ ์๋์ง ์ฒดํฌํด ์คฌ์ผ๋ฉฐ numeric(์ซ์ํ)๊ณผ catergorical(๋ฒ์ฃผํ) ๋ณ์์ ๋ณ๋ณ๋ ฅ์ ์ํด ๊ฐ ๋ณ์์ ํํ๋ฅผ ์ฒดํฌ ํ ๋ฉํ ๋ฐ์ดํฐ ๊ธฐ์ค์ผ๋ก ๊ฐ ๋ณ์๋ฅผ ์ฌ์ ์ํด ์คฌ๋ค.
์ด๋ฐ ์ ์ฒ๋ฆฌ์ ๋ฐ์ดํฐ ์ฒดํฌ ๊ณผ์ ์ ์๊ฐ์ ๋ค์ฌ ๊น๊ฒ ์งํํ ๊ฐ์น๊ฐ ์๋ค. ์ข์ ๋ฐ์ดํฐ๋ฅผ ์ธํํด์ผ ์ข์ ์์ํ์ด ๋์จ๋ค. ์ด๋ค ๋ฐ์ดํฐ๋ฅผ ์ํ๋ ๊ฐ์ผ๋ก ๊ฐ๊ณตํด์ฃผ๋ y = f(x)๋ผ๋ ๋ฐฉ์ ์์์ f๋ผ๋ ์๊ณผ x๋ผ๋ ๋ณ์๊ฐ ๋ชจ๋ ์ณ์์ผ๋ง ์ฌ๋ฐ๋ฅธ y๊ฐ์ ์ป์ด๋ผ ์ ์๋ฏ์ด.
์ด๋ฒ ์คํ ์๋ ๊ธฐ์กด ์ ์ฒ๋ฆฌ๋ ๋ฐ์ดํฐ๋ฅผ ํ์ฉํด 'ํ์ฌ ๊ฐ์ง๊ณ ์๋ ๋ฐ์ดํฐ์ ์์ค์ผ๋ก ๋ฌธ์ ํด๊ฒฐ์ ๋๋ฌํ ์ ์๋์ง ์ ๊ฒ'ํ๊ณ '๋ชจ๋ธ๋ง์ด ์ข์ ์ฑ๋ฅ์ ๋ผ ์ ์๋๋ก ๊ฐ ๋ณ์ ์ฌ์ด์ ๊ด๊ณ๊ฐ ๋ช ํํด์ง๊ฒ ๊ฐ๊ณต/ ์ ๋ณํ๊ฑฐ๋ ๋ฌธ์ ํด๊ฒฐ์ ๋์์ด ๋ ์ ์๋ ๋ณ์๋ฅผ ์ถ๊ฐ๋ก ์์ฑ'ํ๋ Data Readiness Check์ Feature Engineering ๋จ๊ณ๋ฅผ ์งํํ ๊ฒ์ด๋ค.
Data Readiness Check
์๋ก ์์ ์ธ๊ธํ๋ฏ์ด, Data Readiness Check๋ ๊ธฐ์กด ๋ฐ์ดํฐ์ ์์ค์ผ๋ก ๋ฌธ์ ํด๊ฒฐ์ ๋๋ฌํ ์ ์๋์ง ์ ๊ฒํ๋ ํํธ๋ค. ์ด ๊ณผ์ ์์, ์ฐ๋ฆฌ๋ Target label๊ณผ Target Ratio๋ฅผ ์ ๊ฒํด ์ฃผ๊ณ ํด๋น ๋ฐ์ดํฐ๋ฅผ ํ ๋๋ก ๋ถ์ ๋ฐฉํฅ์ฑ์ ์ ์งํ๊ฑฐ๋ ๋ณ๊ฒฝํ ๊ฒ์ด๋ค.
์ด ๊ณผ์ ์ ์ฌ๋ฌ ๊ด์ ์์ ์ ์๋ฏธํ๋ค. ๋ฌธ์ ํด๊ฒฐ ํ๋ก์ธ์ค๋ฅผ ์ ์ํ๋ค๊ณ ํ๋๋ผ๋, ์ง๊ธ ์งํ ๋จ๊ณ์์ Target label์ด ๋์ค์ง ์์ ๊ฒฝ์ฐ๋ ์์ ์ ์๋ค. ์๋ฅผ ๋ค์ด ์ปค๋จธ์ค์ฌ์์ ์ฒซ ๊ตฌ๋งคํ ๊ณ ๊ฐ์ ์ฌ๊ตฌ๋งค ๋น์จ์ ์ธก์ ํ๋ค๊ณ ํ๋ฉด ํด๋น ๋จ๊ณ์์๋ ์ฒซ ๊ตฌ๋งค ๊ณ ๊ฐ ๋ฐ์ดํฐ๋ง ๋์ ์๊ณ ์ฌ๊ตฌ๋งค ๋ฐ์ดํฐ๋ ์์ง ๋์ ์์ง ์์ ์ ์๋ค. ์ด๋ฐ ๊ฒฝ์ฐ์๋ ๋ฐ์ดํฐ ์์ง๋ถํฐ ์ง๊ณ ๊ฐ์ผ ํ๋ค.
์กฐ๊ธ ๋ค๋ฅธ ๋ฐฉํฅ์์ Target Ratio ์์ฒด๋ ์ ์ง๋ง ์์ง ๋ฐ์ดํฐ ์๋ ์์ฒด๋ ๋ง์ ์ ์๋ค. ์ด๋ด ๋๋ ๋ฐ์ดํฐ๋ฅผ ์กฐ๊ธ 'ํฌ๊ธฐํ๊ณ ' ์ผ๋ถ ์ง๋จ์ผ๋ก ํํฐ๋ฅผ ๊ฑธ์ด์ ์ ์ฒด ๋ฐ์ดํฐ๋ฅผ ์ค์ด๋ Target Ratio๋ฅผ ์ฌ๋ฆฌ๋ ๋ฐฉ์์ผ๋ก ์งํํด๋ณผ ์๋ ์๋ค. ์ฃผ ํ๊ฒ์ธต์ด 30~40๋๋ผ๋ฉด 20๋๋ ๋ฐ์ดํฐ์์ ๋นผ๋ฒ๋ฆฌ๊ณ ๋ถ์ํ๋ ๊ฒ์ด๋ค.
๋ ๋ฐ์ดํฐ ๊ฐ์ด ์ผ๋ถ๋ถ์๋ง ์ง๋์น๊ฒ ๋ชฐ๋ ค ์๋ ๊ฒฝ์ฐ๋ ์์ ์ ์๋ค. ์ด๋ฐ ๊ฒฝ์ฐ์๋ ๋ถ์์ด ๋ฌด์๋ฏธํ๋ค. Data Readiness Check๋ ํ๊ฒ ๋ฐ์ดํฐ์ ๋ํด ํ ๋ฒ ๋ ๊ฒ์ ๊ณผ์ ์ ๊ฑฐ์น๊ฒ ํจ์ผ๋ก์จ ์ฐ๋ฆฌ๋ก ํ์ฌ๊ธ ์ด๋ฐ ๋ ผ๋ฆฌ์์ ํ์ ์ ๋ณด๊ฐํ๊ฒ ํด ์ค๋ค.
๋ณธ๊ฒฉ์ ์ผ๋ก ํด๋น ๋จ๊ณ๋ฅผ ์งํํด ๋ณด์. ์ผ๋จ ๋ฐ์ดํฐ๊ฐ ์ด๋ป๊ฒ ์๊ฒผ๋์ง ๋ค์ ํ ๋ฒ ๋ณด๋ฉด,
์ฌ๋ฌ ์งํ๋ค์ด ๋์จ๋ค. ์ด ์ค์ ํ๊ฒ ์งํ๊ฐ ๋๋ WAGE๋ฅผ ์กฐ๊ธ ๋ ๊น๊ฒ ๋ณด์.
describe() ๋ก ๊ธฐ์ด์ ์ธ ๋ณ์์ ๊ธฐ์ ํต๊ณ ์๋ฃ๋ฅผ ์ด๋ํ ์ ์๋ค. distplot ๊ทธ๋ํ์ ํจ๊ป ํ๋ฅผ ํ๋ฒ ๋์ดํ๋ฉด ์๋์ฒ๋ผ ์ถ๋ ฅ๋๋ค.
ํฌ๊ฒ ๋ญ์ณ ์๋ ๋ถ๋ถ ์์ด ์ฐ์ ๋ณ์์ค๋ฌ์ด ๊ทธ๋ํ๋ฅผ ๊ฐ์ง๊ณ ์๋ค. ์ด์ ๊ฐ ๋ณ์์ WAGE ์ฌ์ด์ ๊ด๊ณ๋ฅผ ์กฐ๊ธ ํ์ ํด๋ณผ ์๊ฐ์ด๋ค. groupby๋ก OCCUPATION์ ๊ฐ ๊ฐ๋ณ WAGE์ ํ๊ท ์ ๋ด๋ณด๋ฉด ์๋์ ๊ฐ๋ค. ์ด์ธ์๋ ์ฌ๋ฌ ๊ฐ ์ ๋ ฅํด๋ณด๋ฉด์ ๊ด๊ณ๋ฅผ ์ฐพ์ ๋ณธ๋ค.
์ง๊ธ๊น์ง ํ์ ํ ์ ๋ณด๊ฐ์ผ๋ก ๋ณด๋ฉด, ํด๋น ๋ฐ์ดํฐ๋ก ๋ถ์์ ์งํํด๋ ๋ฌธ์ ์์ด ๋ณด์ธ๋ค. ๋ถํฌ๊ฐ ๊ฝค ๋๊ฒ ํผ์ ธ ์๊ณ , ์ ์๋ฏธํ ๋ณ์๋ ๋ช ๋ฝ์์ผ๋๊น. ๋ค๋ง ์ด์ง ๊ฑฑ์ ๋๋ ์ ์ ๊ฐ์ด 534๊ฐ๋ฐ์ ์๋ค๋ ์ ์ด๋ค. ์ถ๊ฐ์ ์ผ๋ก ์๋ฃ๋ฅผ ๋ ์์งํ ํ์๋ ์๋ค.
Feature Engineering: ๊ฐ์
์์ ์ ์ํ ๋๋ก Feature Engineering์ '๋ชจ๋ธ๋ง์ด ์ข์ ์ฑ๋ฅ์ ๋ผ ์ ์๋๋ก ๊ฐ ๋ณ์ ์ฌ์ด์ ๊ด๊ณ๊ฐ ๋ช ํํด์ง๊ฒ ๊ฐ๊ณต/ ์ ๋ณํ๊ฑฐ๋ ๋ฌธ์ ํด๊ฒฐ์ ๋์์ด ๋ ์ ์๋ ๋ณ์๋ฅผ ์ถ๊ฐ๋ก ์์ฑ'ํ๋ ๊ณผ์ ์ด๋ค. ์ฆ, ์ฌ๊ธฐ์ ๊ฐ์ฅ ๋ฉ์ธ์ด ๋๋ ๋ถ๋ถ์ ์ผ๋ง๋ ๋ ์นด๋ก์ด Feature, ์ฆ ํ๊ฒ ๋ณ์๋ฅผ ๋ง๋ค๊ฑฐ๋ ๊ฐ๊ณตํ๋๋๋ค. ๊ทธ๋ผ ์ด๋ค ๋ฐฉ์์ผ๋ก ์ด Feature๋ฅผ ๋ค๋ฌ์ด์ค ์ ์์๊น? ์ด๋ ํฌ๊ฒ ๋ ๊ฐ์ง ๋ฐฉ์์ผ๋ก ๋๋๋ค.
* Classification(๋ถ๋ฅ) : bin(ํต)์ผ๋ก ๊ตฌ๋ถ ํ Target ๋ณ์์์ ๊ด๊ณ ํ์
* Regression(ํ๊ท) : Target ๋ณ์์์ correlation์ ํ์ธํ๊ฑฐ๋ bin(ํต)์ผ๋ก ๊ตฌ๋ถ ํ Target ๋ณ์์์ ๊ด๊ณ ํ์
์ด์ค Classification, ๋ถ๋ฅ ๋ฐฉ์์ OCCUPATION๊ณผ WAGE ๊ฐ์ ์๊ด๊ด๊ณ๋ฅผ ๊ฐ ๋ณ(bin)๋ก ํ๋ก ์ ๋ฆฌํ๋ ๊ทธ ๋ฐฉ์์ด๋ค. OCCUPATION์ด ๋ฌธ์์ด ๋ณ์์ด๊ณ WAGE์ด ์ฐ์ํ ๋ณ์์ด๊ธฐ ๋๋ฌธ์ ๊น๊ฒ ๋ถ์ํด๋ณผ ์ ์์๋ค. ์๋์๋ ์ฐ์ํ ๋ณ์์ ์ฐ์ํ ๋ณ์์์ ๋น๊ต๋ฅผ Regression, ํ๊ท ๋ฐฉ์์ผ๋ก ์งํํด ๋ณด๊ฒ ๋ค.
Feature Engineering: Numerical features
Numerical feature๋ถํฐ ๋ค๋ค ๋ณธ๋ค. ์ฐ์ ํจํค์ง๋ฅผ ์ค์นํ๊ณ ,
seaborn ํจํค์ง์ heatmap plot์ผ๋ก x์ y๊ฐ ๊ฐ๊ฐ ๋ง๋๋ ์ง์ ์์ ์๊ด๊ณ์๊ฐ ์ด๋ป๊ฒ ๋ผ ์๋์ง ์๊ฐํ๋ ํํ๋ก ํ์ ํด ๋ณด๊ฒ ๋ค.
corr ๋ ์ฝ๋ด๋ ์ด์ ์ด๋ค. ์๊ด ๊ณ์๋ฅผ ๊ตฌํด์ฃผ๋ ๊ฐ์ด๊ณ ์ฐ์ด์ด ๋์ค๋ ๊ฒ๋ค์ ์๊ฐํ ์ต์ ๋ค์ด๋ค.
vmin/ vmax๋ ๋ฒํฐ์ปฌ ๋ ์ธ์ง๋ฅผ ์ ํด ์ฃผ๋ ๊ฐ์ด๋ค. ์๊ด๊ณ์์ ๋ ์ธ์ง๋ฅผ -1๋ถํฐ 1๊น์ง ์ค์ ํ๋ ๊ฒ.
annot์ annotation true๋ก ์ค์ ์๊ด๊ณ์์ ๋ํ ์ซ์๋ฅผ ๊ทธ๋ํ ์์ ์ธ ๊ฑฐ๋ ๋ง ๊ฑฐ๋ ๋ป์ด๋ค. ์ฐ๋ ๊ฒ ๋ ์ง๊ด์ ์ด๊ธฐ ๋๋ฌธ์ ์ฌ๊ธฐ์๋ ํ์ํ๋ค.
mask = mask๋ ์ค๋ณต ๊ฐ์ ๊ทธ๋ํ์์ ์ ๊ฑฐํ๋ ์์ด๋ค. ๋ง์ฝ์ ์ ๊ฑฐํ์ง ์์ผ๋ฉด ์๋ฐฉํฅ์ผ๋ก ๋์ผํ ๊ฐ์ด ๋ฐ์นผ์ฝ๋ง๋์ฒ๋ผ ๊ธฐ์ ๋๋ค.
ํ๋ฅผ ๋ณด์. 0.2 ์ด์์ด๋ฉด ๋ณดํต ์ด๋ ์ ๋ ์๊ด๊ด๊ณ๊ฐ ์๋ค๊ณ ๋ณธ๋ค. ์ด๋ฅผ ๋๊ธด ๊ฐ์ด ๋ ๊ฐ, ์ด์ง ๋ ๋๋ฌํ ๊ฐ ํ๋๊ฐ ๋ณด์ธ๋ค. ๋นจ๊ฐ๊ฒ ๋ค์ด๊ฐ ๋ถ๋ถ์ ๋ณด๋ฉด AGE์ EXPERIENCE์ ๊ด๊ณ๋ค. ๋์ด๊ฐ ๋ง์ ์๋ก ๊ฒฝ๋ ฅ์ด ๋ง๋ ๊ฒ์ธ๋ฐ ์ถ๊ฐ์ ์ธ ํด์์ ์ํ์ง ์๋ ๋จ์ํ ์ ๋ณด๊ฐ์ด๋ค. ์กฐ๊ธ ๋ ๋ณผ๋งํ ์ ๋ณด๋ AGE์ WAGE ๊ฐ์ด๋ค. ๋์ด์ ๋ฐ๋ผ ๊ฒฝ๋ ฅ์ด ๋ง๋ค๋ฉด ๋์ด์ ์ฐ๋ด๋ ์๊ด๊ด๊ณ๊ฐ ์์ ๊ฒ ๊ฐ์๋๋ฐ ์ ์๋ฏธํ ๊ฐ์ ๋๋ฌํ์ง ๋ชป ํ๋ค. ์ด๋ฆฌ๊ณ ์ค๋ ฅ ์ข์ ์ฌ๋๋ค์ด ๋ง์ ๋์ ๋ฐ๊ณ ์๋จ ์ฌ์ค์ ์ฆ์ฏค ๋๊ฒ ๋ค. ๋ค์์ EDUCATION๊ณผ WAGE์ ๊ด๊ณ. ์ด๊ฑด ํ์คํ ์๊ด๊ด๊ณ๊ฐ ์์ด ๋ณด์ธ๋ค. ์กฐ๊ธ ๋ ๊น๊ฒ ๋ณผ๊น.
scatter plot, ์ฐ์ ๋ ํ์์ผ๋ก EDUCATION๊ณผ WAGE์ ๊ด๊ณ๋ฅผ ๊ทธ๋ ค ๋ดค๋ค. EDUCATION์ด ์ฌ๋ผ๊ฐ์๋ก ์ ์ฒด ๊ฐ์ด ์ฐ์ํฅํ๊ณ ์๋ ์ ์ด ๋ณด์ธ๋ค. ์ด ๋ฐ์ดํฐ๋ ๋ถ์์ ์ ์๋ฏธํ ๋ณ์๋ผ ํ ์ ์๊ฒ ๋ค. quantile๋ก ํ๋ก๋ ๋ด ๋ณด์.
๊ฐ์ด ์ ํํ๋ ๊ตฌ๊ฐ์ด ๋ช ์๋ค. 3๊ฐ ์ ๋์ ๊ตฌ๊ฐ์ผ๋ก ๋๋ ๋ณผ ์ ์์ ๊ฒ ๊ฐ๋ค. ์๋ณธ์ ๊ทธ๋๋ก ๋๊ณ ๋ฐ์ดํฐ ๋งํธ๋ฅผ ์ถ๊ฐ์ ์ผ๋ก ๋ง๋ค์ด์ ๊ตฌ๊ฐํํด ๋ณด์.
์ธ ๊ทธ๋ฃน์ ๋น์จ์ ๋ณด์.
0.38์ด๋ผ๋ ์๊ด๊ณ์๋ก๋ ์ผ๋ง๋ ์ค๋ฅด๋์ง ๊ฐ์ด ์ ์๋ ๋ถ๋ถ๋ ๊ทธ๋ฃนํํด์ ๋ฝ์ ๋ณด๋ฉด ๋ช ํํ๊ฒ ๋ณด์ธ๋ค. ๊ทธ๋ผ, ์๊ด๊ด๊ณ๊ฐ 0.18์ ๋์๋ AGE์ WAGE์ ๊ด๊ณ๋ ๊ถ๊ธํด์ง๋ค. ํ ๋ฒ๋ง ๋ ํด๋ณด์.
์ ์๋ฏธํ ๊ด๊ณ๊ฐ ๋ณด์ด์ง ์๋๋ค๋ ๊ฒ์ ์ฝ์ ์ ์๋ค.
์ด์ฒ๋ผ ์ฌ๋ฌ ๊ฐ์ง ๋ฐฉ๋ฒ๋ค๋ก ์๋ฏธ๊ฐ ์๋ ๋ณ์๋ฅผ ์ฐพ์๋ด๊ฑฐ๋ ์๋ฏธ๊ฐ ์๋ ๋ณ์๋ฅผ ๊ฑธ๋ฌ ์ฃผ๋ ์์ ์ ํตํด ๋ฐ์ดํฐ์ ํ๋ฆฌํฐ๋ฅผ ๋ณด๊ฐํ๊ฑฐ๋ ๋ถ์ ๋ฐฉํฅ์ฑ์ ์์ ํ ์ ์๋ค.
Feature Engineering: Catergorical features
Catergorical feature๋ (Numerical feature๋งํผ ๋ณผ๋ฅจ์ด ํฌ์ง ์์ง๋ง) ๊ทธ๋ํ, ํ ๋ฑ์ผ๋ก ํ์ ํ ์ ์๋ค. ์ด๋ ๋ถ์ ๋ชฉ์ ๋ณด๋ค๋ ์ฐ๋ฆฌ๊ฐ ๊ฐ์ง๊ณ ์๋ ์ฐ์ํ ๋ณ์๊ฐ ์ด๋ป๊ฒ ๊ตฌ์ฑ๋์ด ์๋์ง๋ฅผ ๋ณด๊ธฐ ์ํ ๋ชฉ์ ์ด ๋ ํฌ๋ค.
*ํด๋น ํฌ์คํธ๋ ํจ์คํธ์บ ํผ์ค์ ๋ฐ์ดํธ ๋๊ทธ๋ฆฌ EXPORT ๊ฐ์์์ ์ ๊ณต๋๋ ๋ฐ์ดํฐ์ ๋ฐ ์๋ฃ๋ก ์์ฑ๋๋ค.
'TOOLS > Python' ์นดํ ๊ณ ๋ฆฌ์ ๋ค๋ฅธ ๊ธ
Python ๋ชจ๋ธ๋ง: ํ๊ท(Regression) / 1step. ๋ฐ์ดํฐ ์ ์ฒ๋ฆฌ ๋ฐ ํ์์ ๋ฐ์ดํฐ ๋ถ์(EDA) (0) | 2024.02.01 |
---|