월간 데이콘 심리 성향 예측 AI 경진대회

알고리즘 | 정형 | 분류 | 심리 | AUC

Prize : 100만원+애플워치
2,100명 마감

Practice

Overview Data Code (share) Talk Leaderboard

Submission

Private 7위 , Public 32위 0.78206 catboost+nn_model

앵남쓰

2020.11.18 16:00 8,289 Views language

Private 7위 점수: 0.77956, Public 32위 점수 :0.78206,  Catboost  model + NN_model (Ensemble learning)

코드공유 페이지에 Junho Sun님이 올려주신 nn_model입니다 도움이 많이 되었습니다.
nn_model 코드는 밑에 주소에 있습니다.
https://dacon.io/competitions/official/235647/codeshare/1758?page=1&dtype=recent&ptype=pub

위코드에서 마지막 부분만 수정 했습니다.
df = pd.read_csv('/kaggle/input/train-test-sub/sample_submission.csv')
df.iloc[:, 1:] = prediction
df.to_csv("nn_models.csv", index =0)

nn_model 모델  제출 했을 때 점수: 0.7803
cat_boost 모델  제출 했을 때 점수: 0.7780

nn_model + cat_boost (ensemble learning )
# nn_model + ensemble model 다시 올렸습니다.
아래는 제가 작업한 catboost  + nn_model모델 입니다.  kaggle  notebook에서 작업했습니다.
감사합니다.

Code

Comments 3개

로그인이 필요합니다

comment

0 / 1000

앵남쓰

2020.11.18 16:05

# 제가 잠깐 테스트 해본다고 GPU를 지웠는데
# 학습할 때 GPU를 키고 학습하셔야 합니다.
cat = CatBoostClassifier(learning_rate=0.05, iterations=5000, task_type="GPU")
cat.fit(train2, train["voted"], early_stopping_rounds=100, cat_features=['age_group', 'gender', 'race', 'religion'] ,verbose=500)
result = cat.predict_proba(test2)