오늘은 간단하게 발로란트에 있는 무기 스탯에 따라 자동으로 무슨 무기 인지 맞춰주는 모델을 만들어보겠다.
먼저 다음과 같이 임포트 해주고, csv파일(발로란트 총 스탯이 저장된 파일)을 불러와준다.
그리고 X와 Y의 범위를 지정해준 뒤 프린트해서 어떻게 나오는지 한 번 봐준다.
다음과 같이 나오는 것을 볼 수 있다.
그런데 Y가 이상하게 나오니 pd.get_dummies를 이용해준다.
앞에서 했던것과 다르게 나오는 것을 확인할 수 있다.
그리고 다음과 같이 모델을 생성해주고, 실행시켜주면?
이렇게 나온다. 그리고 epoch를 1000번으로 하여
정확도가 높은것을 확인할 수 있다.
이렇게 간단히 총 스탯을 보고 무기 종류를 맞추는 모델을 만들어보았다.