โœ…AutoGluon

AutoML for Image, Text, Time Series, and Tabular Data

AutoGluon์€ ์ด๋ฏธ์ง€, ํ…์ŠคํŠธ, ์‹œ๊ณ„์—ด ๋ฐ ํ…Œ์ด๋ธ” ๋ฐ์ดํ„ฐ์— ๋Œ€ํ•œ ์ž๋™ ๋จธ์‹  ๋Ÿฌ๋‹์„ ์ œ๊ณตํ•˜๋Š” ์†Œํ”„ํŠธ์›จ์–ด์ž…๋‹ˆ๋‹ค. AutoGluon์€ ๋ช‡ ์ค„์˜ ์ฝ”๋“œ๋กœ ๋†’์€ ์ •ํ™•๋„๋ฅผ ๊ฐ€์ง„ ๋จธ์‹  ๋Ÿฌ๋‹ ๋ฐ ๋”ฅ ๋Ÿฌ๋‹ ๋ชจ๋ธ์„ ํ•™์Šตํ•˜๊ณ  ๋ฐฐํฌํ•  ์ˆ˜ ์žˆ๋„๋ก ํ•ด์ค๋‹ˆ๋‹ค.

์ฃผ์š” ํŠน์ง•์€ ๋‹ค์Œ๊ณผ ๊ฐ™์Šต๋‹ˆ๋‹ค.

  • ์ด๋ฏธ์ง€, ํ…์ŠคํŠธ, ์‹œ๊ณ„์—ด ๋ฐ ํ…Œ์ด๋ธ” ๋ฐ์ดํ„ฐ๋ฅผ ํฌํ•จํ•œ ๋‹ค์–‘ํ•œ ๋ฐ์ดํ„ฐ ์œ ํ˜•์— ๋Œ€ํ•ด ์ž๋™ ๋จธ์‹  ๋Ÿฌ๋‹์„ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค.

  • ๊ฐ„๋‹จํ•œ ๋ช‡ ์ค„์˜ ์ฝ”๋“œ๋กœ ๋†’์€ ์ •ํ™•๋„์˜ ๋จธ์‹  ๋Ÿฌ๋‹ ๋ฐ ๋”ฅ ๋Ÿฌ๋‹ ๋ชจ๋ธ์„ ํ•™์Šตํ•˜๊ณ  ๋ฐฐํฌํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.

  • ํŠน์ • ๋ฐ์ดํ„ฐ ์„ธํŠธ์— ๋Œ€ํ•œ ์ตœ์ƒ์˜ ๋ชจ๋ธ์„ ์„ ํƒํ•˜๋Š”๋ฐ ์žˆ์–ด์„œ ์ž๋™ํ™”๋œ ๊ธฐ๋Šฅ์„ ์ œ๊ณตํ•ฉ๋‹ˆ๋‹ค.

  • AutoGluon์€ ์ตœ์ ํ™”๋ฅผ ์œ„ํ•ด ์‹œ์Šคํ…œ ์ž์›์„ ํšจ์œจ์ ์œผ๋กœ ์‚ฌ์šฉํ•˜๋„๋ก ์„ค๊ณ„๋˜์–ด์žˆ์Šต๋‹ˆ๋‹ค.

  • AutoGluon์€ Python 3.8 ์ด์ƒ์—์„œ ์ž‘๋™ํ•˜๋ฉฐ, Apache 2.0 ๋ผ์ด์„ ์Šค๋ฅผ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค.

Example

# First install package from terminal:
# pip install -U pip
# pip install -U setuptools wheel
# pip install autogluon  # autogluon==0.7.0

from autogluon.tabular import TabularDataset, TabularPredictor
train_data = TabularDataset('https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv')
test_data = TabularDataset('https://autogluon.s3.amazonaws.com/datasets/Inc/test.csv')
predictor = TabularPredictor(label='class').fit(train_data, time_limit=120)  # Fit models for 120s
leaderboard = predictor.leaderboard(test_data)

Last updated