{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.10.13","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"none","dataSources":[{"sourceId":80687,"databundleVersionId":8672016,"sourceType":"competition"}],"dockerImageVersionId":30698,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":false}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"markdown","source":"- 이 코드는 완벽하지 않아요\n- roc-auc 평가지표일 때 predict과 predict_proba를 비교해보세요\n- 검증데이터를 사용해보세요!","metadata":{}},{"cell_type":"code","source":"import pandas as pd\ntrain = pd.read_csv(\"/kaggle/input/big-data-analytics-certification-kr-2024-2/train.csv\")\ntest = pd.read_csv(\"/kaggle/input/big-data-analytics-certification-kr-2024-2/test.csv\")","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","execution":{"iopub.status.busy":"2024-05-27T09:28:50.686803Z","iopub.execute_input":"2024-05-27T09:28:50.687305Z","iopub.status.idle":"2024-05-27T09:28:55.855838Z","shell.execute_reply.started":"2024-05-27T09:28:50.687264Z","shell.execute_reply":"2024-05-27T09:28:55.854674Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"from sklearn.ensemble import RandomForestClassifier\n\n# 입력 특성과 타겟 변수 분리\nX_train = train.drop(['id', 'Class'], axis=1)\ny_train = train['Class']\nX_test = test.drop(['id'], axis=1)\n\n# 로지스틱 회귀 모델 학습\nmodel = RandomForestClassifier()\nmodel.fit(X_train, y_train)\n\n# 테스트 데이터에 대한 예측\ny_pred = model.predict(X_test)\n\n# 제출 파일 생성\nsubmission = pd.DataFrame({'id': test['id'], 'Class': y_pred})\nsubmission.to_csv('submission.csv', index=False)","metadata":{"execution":{"iopub.status.busy":"2024-05-27T10:05:48.974891Z","iopub.execute_input":"2024-05-27T10:05:48.975289Z"},"trusted":true},"execution_count":null,"outputs":[]}]}