first commit
This commit is contained in:
Jesselyn Mu
@ -237,6 +237,18 @@
|
||||
"execution_count": 2,
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"df = df.drop(columns=['active_work_category', 'work_stability_score', \n",
|
||||
" 'position_score', 'job_income_position_score',\n",
|
||||
" 'education_score', 'education_income_ratio',\n",
|
||||
" 'weighted_satisfaction_performance'])"
|
||||
]
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 5,
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"X = df.drop(columns=['active_work_months','churn_status', 'employee_id', 'date_of_birth', 'join_date', 'resign_date'])\n",
|
||||
"y = df['active_work_months']\n",
|
||||
@ -253,7 +265,7 @@
|
||||
"y_valid = valid_data['active_work_months']\n",
|
||||
"\n",
|
||||
"cat_feature = ['departemen', 'position', 'domisili', 'marriage_stat', 'job_satisfaction', 'performance_rating',\n",
|
||||
" 'education', 'active_work_category', 'jenis_kelamin']"
|
||||
" 'education', 'jenis_kelamin']"
|
||||
]
|
||||
},
|
||||
{
|
||||
@ -334,21 +346,21 @@
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 5,
|
||||
"execution_count": 6,
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
"name": "stdout",
|
||||
"output_type": "stream",
|
||||
"text": [
|
||||
"0:\tlearn: 14.5009742\ttest: 20.7509764\tbest: 20.7509764 (0)\ttotal: 299ms\tremaining: 4m 58s\n",
|
||||
"200:\tlearn: 2.3680725\ttest: 3.7582712\tbest: 3.7582712 (200)\ttotal: 23.4s\tremaining: 1m 33s\n",
|
||||
"400:\tlearn: 0.6074381\ttest: 1.2438678\tbest: 1.2438678 (400)\ttotal: 39s\tremaining: 58.2s\n",
|
||||
"600:\tlearn: 0.3585168\ttest: 0.6968303\tbest: 0.6968303 (600)\ttotal: 50.6s\tremaining: 33.6s\n",
|
||||
"800:\tlearn: 0.3105016\ttest: 0.5057286\tbest: 0.5057286 (800)\ttotal: 1m 7s\tremaining: 16.7s\n",
|
||||
"999:\tlearn: 0.2918692\ttest: 0.4375722\tbest: 0.4375722 (999)\ttotal: 1m 26s\tremaining: 0us\n",
|
||||
"0:\tlearn: 14.5012046\ttest: 20.7328222\tbest: 20.7328222 (0)\ttotal: 274ms\tremaining: 4m 33s\n",
|
||||
"200:\tlearn: 2.3174890\ttest: 3.6207810\tbest: 3.6207810 (200)\ttotal: 13.5s\tremaining: 53.6s\n",
|
||||
"400:\tlearn: 0.5080249\ttest: 0.7918809\tbest: 0.7918809 (400)\ttotal: 27.5s\tremaining: 41.1s\n",
|
||||
"600:\tlearn: 0.3180844\ttest: 0.3664973\tbest: 0.3664973 (600)\ttotal: 42.6s\tremaining: 28.3s\n",
|
||||
"800:\tlearn: 0.2992983\ttest: 0.3231799\tbest: 0.3231799 (800)\ttotal: 55.5s\tremaining: 13.8s\n",
|
||||
"999:\tlearn: 0.2885793\ttest: 0.3135297\tbest: 0.3135297 (999)\ttotal: 1m 7s\tremaining: 0us\n",
|
||||
"\n",
|
||||
"bestTest = 0.4375721622\n",
|
||||
"bestTest = 0.3135296768\n",
|
||||
"bestIteration = 999\n",
|
||||
"\n"
|
||||
]
|
||||
@ -356,10 +368,10 @@
|
||||
{
|
||||
"data": {
|
||||
"text/plain": [
|
||||
"<catboost.core.CatBoostRegressor at 0x2204d48bd60>"
|
||||
"<catboost.core.CatBoostRegressor at 0x26bbda6b3d0>"
|
||||
]
|
||||
},
|
||||
"execution_count": 5,
|
||||
"execution_count": 6,
|
||||
"metadata": {},
|
||||
"output_type": "execute_result"
|
||||
}
|
||||
@ -549,29 +561,27 @@
|
||||
"name": "stderr",
|
||||
"output_type": "stream",
|
||||
"text": [
|
||||
"d:\\Tugas Akhir\\Codingan\\Development\\App\\.venv\\lib\\site-packages\\tqdm\\auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
|
||||
" from .autonotebook import tqdm as notebook_tqdm\n",
|
||||
"[I 2025-03-20 21:00:35,783] A new study created in memory with name: no-name-90c20206-17ac-4d18-b328-3e83b0b68b02\n",
|
||||
"[I 2025-03-20 21:01:21,239] Trial 0 finished with value: 8.419168492565904 and parameters: {'iterations': 829, 'learning_rate': 0.0015555213398614615, 'depth': 5, 'subsample': 0.7218038924225667, 'colsample_bylevel': 0.6317251084903674, 'l2_leaf_reg': 9.824596206043214, 'random_strength': 5.86501034445993}. Best is trial 0 with value: 8.419168492565904.\n",
|
||||
"[I 2025-03-20 21:02:07,180] Trial 1 finished with value: 2.733330716926934 and parameters: {'iterations': 884, 'learning_rate': 0.0053516318860709, 'depth': 5, 'subsample': 0.7097793314114498, 'colsample_bylevel': 0.6435980070119711, 'l2_leaf_reg': 13.121533573607655, 'random_strength': 9.324119789288343}. Best is trial 1 with value: 2.733330716926934.\n",
|
||||
"[I 2025-03-20 21:02:32,264] Trial 2 finished with value: 1.1897665017142585 and parameters: {'iterations': 635, 'learning_rate': 0.023704951537819915, 'depth': 4, 'subsample': 0.7686389413016427, 'colsample_bylevel': 0.5266671315496564, 'l2_leaf_reg': 15.175290458163712, 'random_strength': 7.30745741132724}. Best is trial 2 with value: 1.1897665017142585.\n",
|
||||
"[I 2025-03-20 21:03:16,738] Trial 3 finished with value: 1.7154507371626437 and parameters: {'iterations': 782, 'learning_rate': 0.012722641891947339, 'depth': 5, 'subsample': 0.6239927785252992, 'colsample_bylevel': 0.551697952957819, 'l2_leaf_reg': 8.011605642650462, 'random_strength': 7.351190134542403}. Best is trial 2 with value: 1.1897665017142585.\n",
|
||||
"[I 2025-03-20 21:03:56,278] Trial 4 finished with value: 1.0342169780611064 and parameters: {'iterations': 590, 'learning_rate': 0.026972969679793426, 'depth': 5, 'subsample': 0.7586099689360802, 'colsample_bylevel': 0.6775428452906951, 'l2_leaf_reg': 8.716335425228655, 'random_strength': 8.009471482166916}. Best is trial 4 with value: 1.0342169780611064.\n",
|
||||
"[I 2025-03-20 21:04:38,974] Trial 5 finished with value: 0.9367588642619448 and parameters: {'iterations': 806, 'learning_rate': 0.025597264277354852, 'depth': 5, 'subsample': 0.6503807167080523, 'colsample_bylevel': 0.6858283170142669, 'l2_leaf_reg': 15.122433690868604, 'random_strength': 9.26015271022034}. Best is trial 5 with value: 0.9367588642619448.\n",
|
||||
"[I 2025-03-20 21:05:17,061] Trial 6 finished with value: 3.3099767746788875 and parameters: {'iterations': 789, 'learning_rate': 0.005334295156738072, 'depth': 4, 'subsample': 0.6150191027518872, 'colsample_bylevel': 0.5100653940971351, 'l2_leaf_reg': 16.26273857150823, 'random_strength': 9.363327197652792}. Best is trial 5 with value: 0.9367588642619448.\n",
|
||||
"[I 2025-03-20 21:06:11,133] Trial 7 finished with value: 2.430297164863871 and parameters: {'iterations': 837, 'learning_rate': 0.006452050997341309, 'depth': 5, 'subsample': 0.5569957978664557, 'colsample_bylevel': 0.7974026616845746, 'l2_leaf_reg': 17.640163162948227, 'random_strength': 8.93735519303648}. Best is trial 5 with value: 0.9367588642619448.\n",
|
||||
"[I 2025-03-20 21:06:40,233] Trial 8 finished with value: 3.059791812580938 and parameters: {'iterations': 590, 'learning_rate': 0.006568464895546362, 'depth': 5, 'subsample': 0.5764687596920144, 'colsample_bylevel': 0.5755754855362132, 'l2_leaf_reg': 8.522648009171846, 'random_strength': 9.538382092752938}. Best is trial 5 with value: 0.9367588642619448.\n",
|
||||
"[I 2025-03-20 21:07:13,639] Trial 9 finished with value: 7.76817781695066 and parameters: {'iterations': 617, 'learning_rate': 0.002539968080520203, 'depth': 4, 'subsample': 0.6372121192487497, 'colsample_bylevel': 0.6082864452448382, 'l2_leaf_reg': 17.654618698706546, 'random_strength': 9.459556159452063}. Best is trial 5 with value: 0.9367588642619448.\n",
|
||||
"[I 2025-03-20 21:08:20,357] Trial 10 finished with value: 0.4103763995321103 and parameters: {'iterations': 979, 'learning_rate': 0.09529892832775133, 'depth': 6, 'subsample': 0.5209516193113186, 'colsample_bylevel': 0.7296584784356337, 'l2_leaf_reg': 5.36555480808817, 'random_strength': 5.055795530221549}. Best is trial 10 with value: 0.4103763995321103.\n",
|
||||
"[I 2025-03-20 21:09:34,792] Trial 11 finished with value: 0.33873805409112234 and parameters: {'iterations': 989, 'learning_rate': 0.07851963900488154, 'depth': 6, 'subsample': 0.5200611359234908, 'colsample_bylevel': 0.7185209948848563, 'l2_leaf_reg': 5.259953652103869, 'random_strength': 5.611812114220319}. Best is trial 11 with value: 0.33873805409112234.\n",
|
||||
"[I 2025-03-20 21:10:53,294] Trial 12 finished with value: 0.33747220251551263 and parameters: {'iterations': 1000, 'learning_rate': 0.09306028108980487, 'depth': 6, 'subsample': 0.5034416033403175, 'colsample_bylevel': 0.7452683981181829, 'l2_leaf_reg': 5.279315402542746, 'random_strength': 5.294692039250562}. Best is trial 12 with value: 0.33747220251551263.\n",
|
||||
"[I 2025-03-20 21:11:51,728] Trial 13 finished with value: 1.6267506473342959 and parameters: {'iterations': 996, 'learning_rate': 0.09325178395683269, 'depth': 6, 'subsample': 0.5125699226468535, 'colsample_bylevel': 0.756377904971589, 'l2_leaf_reg': 5.519312813157406, 'random_strength': 6.120809817278766}. Best is trial 12 with value: 0.33747220251551263.\n",
|
||||
"[I 2025-03-20 21:12:42,167] Trial 14 finished with value: 0.4564504106167195 and parameters: {'iterations': 919, 'learning_rate': 0.052020741738658456, 'depth': 6, 'subsample': 0.5007317413228829, 'colsample_bylevel': 0.727489010605425, 'l2_leaf_reg': 11.083178624062024, 'random_strength': 5.261787256931901}. Best is trial 12 with value: 0.33747220251551263.\n",
|
||||
"[I 2025-03-20 21:13:32,464] Trial 15 finished with value: 0.6231066699606179 and parameters: {'iterations': 696, 'learning_rate': 0.04926421412297825, 'depth': 6, 'subsample': 0.5528336472517523, 'colsample_bylevel': 0.7710394282985423, 'l2_leaf_reg': 6.189751630816335, 'random_strength': 6.502158303659374}. Best is trial 12 with value: 0.33747220251551263.\n",
|
||||
"[I 2025-03-20 21:14:53,119] Trial 16 finished with value: 0.4792492732081974 and parameters: {'iterations': 930, 'learning_rate': 0.04640033657256652, 'depth': 6, 'subsample': 0.5859719980036684, 'colsample_bylevel': 0.700113722637731, 'l2_leaf_reg': 7.099786946755341, 'random_strength': 6.688129574746723}. Best is trial 12 with value: 0.33747220251551263.\n",
|
||||
"[I 2025-03-20 21:15:49,049] Trial 17 finished with value: 1.61841931396652 and parameters: {'iterations': 518, 'learning_rate': 0.015200955659638398, 'depth': 6, 'subsample': 0.534977261704631, 'colsample_bylevel': 0.7246250429139481, 'l2_leaf_reg': 19.80600077744929, 'random_strength': 5.673411846650424}. Best is trial 12 with value: 0.33747220251551263.\n",
|
||||
"[I 2025-03-20 21:17:36,811] Trial 18 finished with value: 0.5548288731588591 and parameters: {'iterations': 938, 'learning_rate': 0.07194118095780964, 'depth': 6, 'subsample': 0.6665507151821397, 'colsample_bylevel': 0.790138958827564, 'l2_leaf_reg': 12.301321434830928, 'random_strength': 6.796168129707886}. Best is trial 12 with value: 0.33747220251551263.\n",
|
||||
"[I 2025-03-20 21:18:43,121] Trial 19 finished with value: 0.6231757616191587 and parameters: {'iterations': 719, 'learning_rate': 0.03682071614516647, 'depth': 6, 'subsample': 0.5866796565002883, 'colsample_bylevel': 0.7504213556109071, 'l2_leaf_reg': 10.442003723130012, 'random_strength': 8.36360703538023}. Best is trial 12 with value: 0.33747220251551263.\n"
|
||||
"[I 2025-04-15 15:08:11,750] A new study created in memory with name: no-name-7d5ebf34-fc55-4b59-b90c-b22b64311951\n",
|
||||
"[I 2025-04-15 15:08:48,891] Trial 0 finished with value: 0.646062322615293 and parameters: {'iterations': 898, 'learning_rate': 0.03145083099289301, 'depth': 4, 'subsample': 0.6444105892248361, 'colsample_bylevel': 0.6186096575774811, 'l2_leaf_reg': 17.64514783498297, 'random_strength': 9.88786820068755}. Best is trial 0 with value: 0.646062322615293.\n",
|
||||
"[I 2025-04-15 15:09:26,486] Trial 1 finished with value: 0.9184440015716631 and parameters: {'iterations': 897, 'learning_rate': 0.011291887138100785, 'depth': 4, 'subsample': 0.7769524548266592, 'colsample_bylevel': 0.5647743432086197, 'l2_leaf_reg': 7.76546189160343, 'random_strength': 7.668666969032969}. Best is trial 0 with value: 0.646062322615293.\n",
|
||||
"[I 2025-04-15 15:09:52,174] Trial 2 finished with value: 5.984384729372321 and parameters: {'iterations': 598, 'learning_rate': 0.004160874913367698, 'depth': 4, 'subsample': 0.5118790113991489, 'colsample_bylevel': 0.5110086574656876, 'l2_leaf_reg': 18.661041022397058, 'random_strength': 5.943307589760212}. Best is trial 0 with value: 0.646062322615293.\n",
|
||||
"[I 2025-04-15 15:10:26,575] Trial 3 finished with value: 9.228750640503423 and parameters: {'iterations': 682, 'learning_rate': 0.0021975421219896157, 'depth': 5, 'subsample': 0.6755358151480091, 'colsample_bylevel': 0.6596729110914161, 'l2_leaf_reg': 14.736656536586944, 'random_strength': 7.136912007339907}. Best is trial 0 with value: 0.646062322615293.\n",
|
||||
"[I 2025-04-15 15:11:10,104] Trial 4 finished with value: 0.6227136479365861 and parameters: {'iterations': 895, 'learning_rate': 0.019358429409891725, 'depth': 5, 'subsample': 0.6733355260936932, 'colsample_bylevel': 0.6148187038915378, 'l2_leaf_reg': 12.647440499487868, 'random_strength': 9.155074074386231}. Best is trial 4 with value: 0.6227136479365861.\n",
|
||||
"[I 2025-04-15 15:11:42,996] Trial 5 finished with value: 2.612658773597153 and parameters: {'iterations': 607, 'learning_rate': 0.0064697585951560485, 'depth': 6, 'subsample': 0.5164557295530797, 'colsample_bylevel': 0.6841029537714532, 'l2_leaf_reg': 11.459664006812375, 'random_strength': 5.575562702191829}. Best is trial 4 with value: 0.6227136479365861.\n",
|
||||
"[I 2025-04-15 15:12:09,055] Trial 6 finished with value: 5.0563199381852675 and parameters: {'iterations': 634, 'learning_rate': 0.0041378039430250224, 'depth': 4, 'subsample': 0.6296012316767893, 'colsample_bylevel': 0.6928068433595623, 'l2_leaf_reg': 18.60249289606864, 'random_strength': 6.29108320180428}. Best is trial 4 with value: 0.6227136479365861.\n",
|
||||
"[I 2025-04-15 15:12:48,070] Trial 7 finished with value: 6.064226235109285 and parameters: {'iterations': 659, 'learning_rate': 0.003194121199679894, 'depth': 6, 'subsample': 0.6850485984896206, 'colsample_bylevel': 0.7100729641171083, 'l2_leaf_reg': 11.897665097253151, 'random_strength': 5.260194361969255}. Best is trial 4 with value: 0.6227136479365861.\n",
|
||||
"[I 2025-04-15 15:13:18,641] Trial 8 finished with value: 9.453945919462235 and parameters: {'iterations': 700, 'learning_rate': 0.0020817037350174448, 'depth': 4, 'subsample': 0.6203198807137237, 'colsample_bylevel': 0.5266772331906592, 'l2_leaf_reg': 9.86926120392112, 'random_strength': 5.474003236157858}. Best is trial 4 with value: 0.6227136479365861.\n",
|
||||
"[I 2025-04-15 15:14:05,709] Trial 9 finished with value: 0.49267024172790747 and parameters: {'iterations': 687, 'learning_rate': 0.052164593502382166, 'depth': 6, 'subsample': 0.6660663261019623, 'colsample_bylevel': 0.7314001039861265, 'l2_leaf_reg': 17.594770060423073, 'random_strength': 9.339230639028976}. Best is trial 9 with value: 0.49267024172790747.\n",
|
||||
"[I 2025-04-15 15:14:39,848] Trial 10 finished with value: 0.5500268252840363 and parameters: {'iterations': 509, 'learning_rate': 0.07967431030832754, 'depth': 6, 'subsample': 0.7602272787841288, 'colsample_bylevel': 0.770229071536223, 'l2_leaf_reg': 15.503851488383855, 'random_strength': 8.492987824560748}. Best is trial 9 with value: 0.49267024172790747.\n",
|
||||
"[I 2025-04-15 15:15:11,853] Trial 11 finished with value: 0.49023595998871994 and parameters: {'iterations': 500, 'learning_rate': 0.08779241979856825, 'depth': 6, 'subsample': 0.796075504979686, 'colsample_bylevel': 0.7976372716940351, 'l2_leaf_reg': 14.93516913511941, 'random_strength': 8.423535620874652}. Best is trial 11 with value: 0.49023595998871994.\n",
|
||||
"[I 2025-04-15 15:16:03,033] Trial 12 finished with value: 0.49819783655018374 and parameters: {'iterations': 793, 'learning_rate': 0.09384534188388771, 'depth': 6, 'subsample': 0.7184900016067328, 'colsample_bylevel': 0.7836196128120625, 'l2_leaf_reg': 15.663609062290409, 'random_strength': 8.589899472344362}. Best is trial 11 with value: 0.49023595998871994.\n",
|
||||
"[I 2025-04-15 15:16:29,620] Trial 13 finished with value: 0.4846695605629031 and parameters: {'iterations': 537, 'learning_rate': 0.042877612832129756, 'depth': 5, 'subsample': 0.5706634178949412, 'colsample_bylevel': 0.7465651513848619, 'l2_leaf_reg': 5.218951437778482, 'random_strength': 9.92877520468897}. Best is trial 13 with value: 0.4846695605629031.\n",
|
||||
"[I 2025-04-15 15:16:57,316] Trial 14 finished with value: 0.7077219444540906 and parameters: {'iterations': 504, 'learning_rate': 0.034174948663442034, 'depth': 5, 'subsample': 0.569558755228507, 'colsample_bylevel': 0.7511709728692211, 'l2_leaf_reg': 7.512089049291371, 'random_strength': 7.737784610626449}. Best is trial 13 with value: 0.4846695605629031.\n",
|
||||
"[I 2025-04-15 15:17:22,465] Trial 15 finished with value: 0.8736557180076023 and parameters: {'iterations': 560, 'learning_rate': 0.017043234915075773, 'depth': 5, 'subsample': 0.5741018674613553, 'colsample_bylevel': 0.7991676085324451, 'l2_leaf_reg': 5.982967530770232, 'random_strength': 9.756085101353683}. Best is trial 13 with value: 0.4846695605629031.\n",
|
||||
"[I 2025-04-15 15:18:03,448] Trial 16 finished with value: 0.45211315741628616 and parameters: {'iterations': 779, 'learning_rate': 0.049661228877482096, 'depth': 5, 'subsample': 0.7989351952288142, 'colsample_bylevel': 0.7407581801522412, 'l2_leaf_reg': 5.003342383992051, 'random_strength': 8.391196012503668}. Best is trial 16 with value: 0.45211315741628616.\n",
|
||||
"[I 2025-04-15 15:18:44,012] Trial 17 finished with value: 12.445283924362869 and parameters: {'iterations': 791, 'learning_rate': 0.0011590379392432504, 'depth': 5, 'subsample': 0.5835792160594921, 'colsample_bylevel': 0.7376020551248276, 'l2_leaf_reg': 5.088499987791439, 'random_strength': 7.132621480885446}. Best is trial 16 with value: 0.45211315741628616.\n",
|
||||
"[I 2025-04-15 15:19:38,776] Trial 18 finished with value: 0.4335016501614797 and parameters: {'iterations': 970, 'learning_rate': 0.04163009683311559, 'depth': 5, 'subsample': 0.7311095765931528, 'colsample_bylevel': 0.6587798610446518, 'l2_leaf_reg': 7.70099272097691, 'random_strength': 9.134434266536488}. Best is trial 18 with value: 0.4335016501614797.\n",
|
||||
"[I 2025-04-15 15:20:27,679] Trial 19 finished with value: 0.639097289338391 and parameters: {'iterations': 979, 'learning_rate': 0.012183824359018583, 'depth': 5, 'subsample': 0.7371061668104383, 'colsample_bylevel': 0.6302275501729142, 'l2_leaf_reg': 8.50453575427487, 'random_strength': 9.030709992017943}. Best is trial 18 with value: 0.4335016501614797.\n"
|
||||
]
|
||||
},
|
||||
{
|
||||
@ -579,8 +589,8 @@
|
||||
"output_type": "stream",
|
||||
"text": [
|
||||
"Best trial:\n",
|
||||
" RMSE: 0.33747220251551263\n",
|
||||
" Params: {'iterations': 1000, 'learning_rate': 0.09306028108980487, 'depth': 6, 'subsample': 0.5034416033403175, 'colsample_bylevel': 0.7452683981181829, 'l2_leaf_reg': 5.279315402542746, 'random_strength': 5.294692039250562}\n"
|
||||
" RMSE: 0.4335016501614797\n",
|
||||
" Params: {'iterations': 970, 'learning_rate': 0.04163009683311559, 'depth': 5, 'subsample': 0.7311095765931528, 'colsample_bylevel': 0.6587798610446518, 'l2_leaf_reg': 7.70099272097691, 'random_strength': 9.134434266536488}\n"
|
||||
]
|
||||
}
|
||||
],
|
||||
@ -588,6 +598,7 @@
|
||||
"import optuna\n",
|
||||
"from catboost import CatBoostRegressor\n",
|
||||
"from sklearn.metrics import mean_squared_error\n",
|
||||
"import numpy as np\n",
|
||||
"\n",
|
||||
"# Fungsi objective untuk Optuna\n",
|
||||
"def objective(trial):\n",
|
||||
@ -639,18 +650,17 @@
|
||||
"name": "stdout",
|
||||
"output_type": "stream",
|
||||
"text": [
|
||||
"0:\tlearn: 13.4115622\ttest: 19.0660791\tbest: 19.0660791 (0)\ttotal: 193ms\tremaining: 3m 13s\n",
|
||||
"200:\tlearn: 0.3985082\ttest: 0.5382811\tbest: 0.5382373 (199)\ttotal: 21.7s\tremaining: 1m 26s\n",
|
||||
"400:\tlearn: 0.2915505\ttest: 0.4082137\tbest: 0.4080268 (398)\ttotal: 38.8s\tremaining: 57.9s\n",
|
||||
"600:\tlearn: 0.2452564\ttest: 0.3555501\tbest: 0.3555501 (600)\ttotal: 55.4s\tremaining: 36.8s\n",
|
||||
"800:\tlearn: 0.2201339\ttest: 0.3435762\tbest: 0.3435629 (786)\ttotal: 1m 16s\tremaining: 18.9s\n",
|
||||
"999:\tlearn: 0.2028099\ttest: 0.3375486\tbest: 0.3374722 (998)\ttotal: 1m 40s\tremaining: 0us\n",
|
||||
"0:\tlearn: 14.2026549\ttest: 20.4309858\tbest: 20.4309858 (0)\ttotal: 111ms\tremaining: 1m 47s\n",
|
||||
"200:\tlearn: 0.7177922\ttest: 1.1667817\tbest: 1.1667817 (200)\ttotal: 10.7s\tremaining: 40.8s\n",
|
||||
"400:\tlearn: 0.4239619\ttest: 0.5853924\tbest: 0.5853924 (400)\ttotal: 22.1s\tremaining: 31.4s\n",
|
||||
"600:\tlearn: 0.3515389\ttest: 0.5013024\tbest: 0.5012549 (597)\ttotal: 32.9s\tremaining: 20.2s\n",
|
||||
"800:\tlearn: 0.3162819\ttest: 0.4551766\tbest: 0.4551445 (798)\ttotal: 44.8s\tremaining: 9.45s\n",
|
||||
"969:\tlearn: 0.2948077\ttest: 0.4335017\tbest: 0.4335017 (969)\ttotal: 53.6s\tremaining: 0us\n",
|
||||
"\n",
|
||||
"bestTest = 0.3374722025\n",
|
||||
"bestIteration = 998\n",
|
||||
"bestTest = 0.4335016502\n",
|
||||
"bestIteration = 969\n",
|
||||
"\n",
|
||||
"Shrink model to first 999 iterations.\n",
|
||||
"Final RMSE: 0.33747220251551263\n"
|
||||
"Final RMSE: 0.4335016501614797\n"
|
||||
]
|
||||
}
|
||||
],
|
||||
@ -950,7 +960,7 @@
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 14,
|
||||
"execution_count": 10,
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
|
||||
Reference in New Issue
Block a user