Pytorch로 텍스트 분류 하기
·
머신러닝/PyTorch
뉴스 카테고리 분류 모델 대회 데이터 다운 링크 private score : 0.67608 , public score : 0.70427 사용 모델 : LSTM 누구나 이해하기 쉽게 가장 간단하게 설계한 모델입니다. 참고사이트 : 위키 독스 import pandas as pd train = pd.read_csv('train_data.csv') test = pd.read_csv('test_data.csv') 간단한 EDA train.head() index title topic_idx 0 0 인천→핀란드 항공기 결항…휴가철 여행객 분통 4 1 1 실리콘밸리 넘어서겠다…구글 15조원 들여 美전역 거점화 4 2 2 이란 외무 긴장완화 해결책은 미국이 경제전쟁 멈추는 것 4 3 3 NYT 클린턴 측근韓기업 특수관계..