Last active
August 11, 2019 14:37
-
-
Save szdr/95d499bbe6d35bda8c5bc386a1639b97 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
{ | |
"cells": [ | |
{ | |
"cell_type": "code", | |
"execution_count": 1, | |
"metadata": {}, | |
"outputs": [ | |
{ | |
"name": "stderr", | |
"output_type": "stream", | |
"text": [ | |
"/usr/local/lib/python3.7/site-packages/lightgbm/__init__.py:46: UserWarning: Starting from version 2.2.1, the library file in distribution wheels for macOS is built by the Apple Clang (Xcode_8.3.3) compiler.\n", | |
"This means that in case of installing LightGBM from PyPI via the ``pip install lightgbm`` command, you don't need to install the gcc compiler anymore.\n", | |
"Instead of that, you need to install the OpenMP library, which is required for running LightGBM on the system with the Apple Clang compiler.\n", | |
"You can install the OpenMP library by the following command: ``brew install libomp``.\n", | |
" \"You can install the OpenMP library by the following command: ``brew install libomp``.\", UserWarning)\n" | |
] | |
} | |
], | |
"source": [ | |
"import lightgbm as lgb" | |
] | |
}, | |
{ | |
"cell_type": "code", | |
"execution_count": 13, | |
"metadata": {}, | |
"outputs": [], | |
"source": [ | |
"dataset_params={\"has_header\": True, \"label_column\": \"name:label\", \"group_column\": \"name:qid\"}" | |
] | |
}, | |
{ | |
"cell_type": "code", | |
"execution_count": 14, | |
"metadata": {}, | |
"outputs": [], | |
"source": [ | |
"train_dataset = lgb.Dataset(\"./lgbm_train.tsv\", params=dataset_params)\n", | |
"test_dataset = lgb.Dataset(\"./lgbm_test.tsv\", params=dataset_params)" | |
] | |
}, | |
{ | |
"cell_type": "code", | |
"execution_count": 15, | |
"metadata": {}, | |
"outputs": [], | |
"source": [ | |
"train_params = {\n", | |
" \"task\": \"train\",\n", | |
" \"boosting_type\": \"gbdt\",\n", | |
" \"objective\": \"lambdarank\",\n", | |
" \"metric\": \"ndcg\",\n", | |
" \"learning_rate\": 0.1,\n", | |
" \"num_leaves\": 31,\n", | |
" \"feature_fraction\": 1.0,\n", | |
" \"bagging_freq\": 1,\n", | |
" \"bagging_fraction\": 0.9,\n", | |
" \"min_data_in_leaf\": 50,\n", | |
" \"min_sum_hessian_in_leaf\": 5.0,\n", | |
"}" | |
] | |
}, | |
{ | |
"cell_type": "code", | |
"execution_count": 16, | |
"metadata": {}, | |
"outputs": [ | |
{ | |
"name": "stdout", | |
"output_type": "stream", | |
"text": [ | |
"[1]\tvalid_0's ndcg@1: 0.528762\tvalid_0's ndcg@2: 0.544562\tvalid_0's ndcg@3: 0.583405\tvalid_0's ndcg@4: 0.613734\tvalid_0's ndcg@5: 0.635418\n", | |
"[2]\tvalid_0's ndcg@1: 0.58019\tvalid_0's ndcg@2: 0.584599\tvalid_0's ndcg@3: 0.606662\tvalid_0's ndcg@4: 0.643052\tvalid_0's ndcg@5: 0.646633\n", | |
"[3]\tvalid_0's ndcg@1: 0.569143\tvalid_0's ndcg@2: 0.582235\tvalid_0's ndcg@3: 0.594071\tvalid_0's ndcg@4: 0.615124\tvalid_0's ndcg@5: 0.653975\n", | |
"[4]\tvalid_0's ndcg@1: 0.579619\tvalid_0's ndcg@2: 0.58708\tvalid_0's ndcg@3: 0.597172\tvalid_0's ndcg@4: 0.615615\tvalid_0's ndcg@5: 0.633487\n", | |
"[5]\tvalid_0's ndcg@1: 0.598667\tvalid_0's ndcg@2: 0.603885\tvalid_0's ndcg@3: 0.598697\tvalid_0's ndcg@4: 0.614908\tvalid_0's ndcg@5: 0.635176\n", | |
"[6]\tvalid_0's ndcg@1: 0.564381\tvalid_0's ndcg@2: 0.593141\tvalid_0's ndcg@3: 0.599503\tvalid_0's ndcg@4: 0.614947\tvalid_0's ndcg@5: 0.631274\n", | |
"[7]\tvalid_0's ndcg@1: 0.573905\tvalid_0's ndcg@2: 0.591795\tvalid_0's ndcg@3: 0.591546\tvalid_0's ndcg@4: 0.597398\tvalid_0's ndcg@5: 0.620378\n", | |
"[8]\tvalid_0's ndcg@1: 0.586286\tvalid_0's ndcg@2: 0.596302\tvalid_0's ndcg@3: 0.596648\tvalid_0's ndcg@4: 0.606899\tvalid_0's ndcg@5: 0.634188\n", | |
"[9]\tvalid_0's ndcg@1: 0.583429\tvalid_0's ndcg@2: 0.570345\tvalid_0's ndcg@3: 0.58036\tvalid_0's ndcg@4: 0.613106\tvalid_0's ndcg@5: 0.625483\n", | |
"[10]\tvalid_0's ndcg@1: 0.583429\tvalid_0's ndcg@2: 0.581234\tvalid_0's ndcg@3: 0.58014\tvalid_0's ndcg@4: 0.612795\tvalid_0's ndcg@5: 0.638266\n", | |
"[11]\tvalid_0's ndcg@1: 0.558667\tvalid_0's ndcg@2: 0.594977\tvalid_0's ndcg@3: 0.588791\tvalid_0's ndcg@4: 0.599909\tvalid_0's ndcg@5: 0.62218\n", | |
"[12]\tvalid_0's ndcg@1: 0.572\tvalid_0's ndcg@2: 0.589478\tvalid_0's ndcg@3: 0.592286\tvalid_0's ndcg@4: 0.600885\tvalid_0's ndcg@5: 0.615476\n", | |
"[13]\tvalid_0's ndcg@1: 0.582476\tvalid_0's ndcg@2: 0.595516\tvalid_0's ndcg@3: 0.59007\tvalid_0's ndcg@4: 0.603868\tvalid_0's ndcg@5: 0.622617\n", | |
"[14]\tvalid_0's ndcg@1: 0.60781\tvalid_0's ndcg@2: 0.589929\tvalid_0's ndcg@3: 0.595379\tvalid_0's ndcg@4: 0.611253\tvalid_0's ndcg@5: 0.626625\n", | |
"[15]\tvalid_0's ndcg@1: 0.60781\tvalid_0's ndcg@2: 0.589562\tvalid_0's ndcg@3: 0.601567\tvalid_0's ndcg@4: 0.604624\tvalid_0's ndcg@5: 0.615493\n", | |
"[16]\tvalid_0's ndcg@1: 0.621143\tvalid_0's ndcg@2: 0.597479\tvalid_0's ndcg@3: 0.604006\tvalid_0's ndcg@4: 0.6179\tvalid_0's ndcg@5: 0.636303\n", | |
"[17]\tvalid_0's ndcg@1: 0.613524\tvalid_0's ndcg@2: 0.605268\tvalid_0's ndcg@3: 0.612371\tvalid_0's ndcg@4: 0.626398\tvalid_0's ndcg@5: 0.642013\n", | |
"[18]\tvalid_0's ndcg@1: 0.593524\tvalid_0's ndcg@2: 0.588654\tvalid_0's ndcg@3: 0.604158\tvalid_0's ndcg@4: 0.619333\tvalid_0's ndcg@5: 0.646776\n", | |
"[19]\tvalid_0's ndcg@1: 0.594476\tvalid_0's ndcg@2: 0.581958\tvalid_0's ndcg@3: 0.604156\tvalid_0's ndcg@4: 0.612915\tvalid_0's ndcg@5: 0.652193\n", | |
"[20]\tvalid_0's ndcg@1: 0.58781\tvalid_0's ndcg@2: 0.607823\tvalid_0's ndcg@3: 0.610994\tvalid_0's ndcg@4: 0.622306\tvalid_0's ndcg@5: 0.656395\n", | |
"[21]\tvalid_0's ndcg@1: 0.60781\tvalid_0's ndcg@2: 0.597925\tvalid_0's ndcg@3: 0.599301\tvalid_0's ndcg@4: 0.621136\tvalid_0's ndcg@5: 0.65195\n", | |
"[22]\tvalid_0's ndcg@1: 0.58781\tvalid_0's ndcg@2: 0.579287\tvalid_0's ndcg@3: 0.596673\tvalid_0's ndcg@4: 0.620999\tvalid_0's ndcg@5: 0.638346\n", | |
"[23]\tvalid_0's ndcg@1: 0.58781\tvalid_0's ndcg@2: 0.576708\tvalid_0's ndcg@3: 0.581329\tvalid_0's ndcg@4: 0.606692\tvalid_0's ndcg@5: 0.641438\n", | |
"[24]\tvalid_0's ndcg@1: 0.582095\tvalid_0's ndcg@2: 0.587913\tvalid_0's ndcg@3: 0.590781\tvalid_0's ndcg@4: 0.607598\tvalid_0's ndcg@5: 0.638129\n", | |
"[25]\tvalid_0's ndcg@1: 0.599238\tvalid_0's ndcg@2: 0.599821\tvalid_0's ndcg@3: 0.60578\tvalid_0's ndcg@4: 0.619131\tvalid_0's ndcg@5: 0.643381\n", | |
"[26]\tvalid_0's ndcg@1: 0.595619\tvalid_0's ndcg@2: 0.591475\tvalid_0's ndcg@3: 0.60474\tvalid_0's ndcg@4: 0.614853\tvalid_0's ndcg@5: 0.63531\n", | |
"[27]\tvalid_0's ndcg@1: 0.575619\tvalid_0's ndcg@2: 0.591371\tvalid_0's ndcg@3: 0.611098\tvalid_0's ndcg@4: 0.620853\tvalid_0's ndcg@5: 0.638049\n", | |
"[28]\tvalid_0's ndcg@1: 0.569905\tvalid_0's ndcg@2: 0.595974\tvalid_0's ndcg@3: 0.602347\tvalid_0's ndcg@4: 0.618284\tvalid_0's ndcg@5: 0.643449\n", | |
"[29]\tvalid_0's ndcg@1: 0.583238\tvalid_0's ndcg@2: 0.598991\tvalid_0's ndcg@3: 0.602841\tvalid_0's ndcg@4: 0.632412\tvalid_0's ndcg@5: 0.642402\n", | |
"[30]\tvalid_0's ndcg@1: 0.609905\tvalid_0's ndcg@2: 0.60392\tvalid_0's ndcg@3: 0.607219\tvalid_0's ndcg@4: 0.628887\tvalid_0's ndcg@5: 0.647752\n", | |
"[31]\tvalid_0's ndcg@1: 0.603238\tvalid_0's ndcg@2: 0.6028\tvalid_0's ndcg@3: 0.60772\tvalid_0's ndcg@4: 0.62196\tvalid_0's ndcg@5: 0.64849\n", | |
"[32]\tvalid_0's ndcg@1: 0.583238\tvalid_0's ndcg@2: 0.594202\tvalid_0's ndcg@3: 0.609342\tvalid_0's ndcg@4: 0.626613\tvalid_0's ndcg@5: 0.637428\n", | |
"[33]\tvalid_0's ndcg@1: 0.588952\tvalid_0's ndcg@2: 0.594111\tvalid_0's ndcg@3: 0.605456\tvalid_0's ndcg@4: 0.615482\tvalid_0's ndcg@5: 0.636358\n", | |
"[34]\tvalid_0's ndcg@1: 0.567048\tvalid_0's ndcg@2: 0.569589\tvalid_0's ndcg@3: 0.593125\tvalid_0's ndcg@4: 0.616653\tvalid_0's ndcg@5: 0.631836\n", | |
"[35]\tvalid_0's ndcg@1: 0.567048\tvalid_0's ndcg@2: 0.573642\tvalid_0's ndcg@3: 0.604459\tvalid_0's ndcg@4: 0.617112\tvalid_0's ndcg@5: 0.644812\n", | |
"[36]\tvalid_0's ndcg@1: 0.561333\tvalid_0's ndcg@2: 0.584601\tvalid_0's ndcg@3: 0.60271\tvalid_0's ndcg@4: 0.619038\tvalid_0's ndcg@5: 0.640994\n", | |
"[37]\tvalid_0's ndcg@1: 0.568\tvalid_0's ndcg@2: 0.599042\tvalid_0's ndcg@3: 0.607998\tvalid_0's ndcg@4: 0.621502\tvalid_0's ndcg@5: 0.645621\n", | |
"[38]\tvalid_0's ndcg@1: 0.587048\tvalid_0's ndcg@2: 0.606829\tvalid_0's ndcg@3: 0.61438\tvalid_0's ndcg@4: 0.633104\tvalid_0's ndcg@5: 0.650324\n", | |
"[39]\tvalid_0's ndcg@1: 0.587048\tvalid_0's ndcg@2: 0.601671\tvalid_0's ndcg@3: 0.61669\tvalid_0's ndcg@4: 0.627179\tvalid_0's ndcg@5: 0.649764\n", | |
"[40]\tvalid_0's ndcg@1: 0.587048\tvalid_0's ndcg@2: 0.584171\tvalid_0's ndcg@3: 0.610716\tvalid_0's ndcg@4: 0.622388\tvalid_0's ndcg@5: 0.64301\n", | |
"[41]\tvalid_0's ndcg@1: 0.587048\tvalid_0's ndcg@2: 0.579909\tvalid_0's ndcg@3: 0.618761\tvalid_0's ndcg@4: 0.627474\tvalid_0's ndcg@5: 0.644692\n", | |
"[42]\tvalid_0's ndcg@1: 0.613714\tvalid_0's ndcg@2: 0.589682\tvalid_0's ndcg@3: 0.618136\tvalid_0's ndcg@4: 0.636468\tvalid_0's ndcg@5: 0.652117\n", | |
"[43]\tvalid_0's ndcg@1: 0.600381\tvalid_0's ndcg@2: 0.581507\tvalid_0's ndcg@3: 0.61078\tvalid_0's ndcg@4: 0.633716\tvalid_0's ndcg@5: 0.651001\n", | |
"[44]\tvalid_0's ndcg@1: 0.580381\tvalid_0's ndcg@2: 0.582139\tvalid_0's ndcg@3: 0.607689\tvalid_0's ndcg@4: 0.628135\tvalid_0's ndcg@5: 0.645777\n", | |
"[45]\tvalid_0's ndcg@1: 0.580381\tvalid_0's ndcg@2: 0.582139\tvalid_0's ndcg@3: 0.603443\tvalid_0's ndcg@4: 0.627613\tvalid_0's ndcg@5: 0.650034\n", | |
"[46]\tvalid_0's ndcg@1: 0.59181\tvalid_0's ndcg@2: 0.574409\tvalid_0's ndcg@3: 0.608843\tvalid_0's ndcg@4: 0.625241\tvalid_0's ndcg@5: 0.647728\n", | |
"[47]\tvalid_0's ndcg@1: 0.59181\tvalid_0's ndcg@2: 0.579567\tvalid_0's ndcg@3: 0.614131\tvalid_0's ndcg@4: 0.630868\tvalid_0's ndcg@5: 0.642405\n", | |
"[48]\tvalid_0's ndcg@1: 0.59181\tvalid_0's ndcg@2: 0.579567\tvalid_0's ndcg@3: 0.619233\tvalid_0's ndcg@4: 0.639314\tvalid_0's ndcg@5: 0.647826\n", | |
"[49]\tvalid_0's ndcg@1: 0.567048\tvalid_0's ndcg@2: 0.57323\tvalid_0's ndcg@3: 0.610322\tvalid_0's ndcg@4: 0.628037\tvalid_0's ndcg@5: 0.641127\n", | |
"[50]\tvalid_0's ndcg@1: 0.537524\tvalid_0's ndcg@2: 0.559405\tvalid_0's ndcg@3: 0.601415\tvalid_0's ndcg@4: 0.621678\tvalid_0's ndcg@5: 0.637118\n", | |
"[51]\tvalid_0's ndcg@1: 0.52419\tvalid_0's ndcg@2: 0.562064\tvalid_0's ndcg@3: 0.594304\tvalid_0's ndcg@4: 0.61995\tvalid_0's ndcg@5: 0.6303\n", | |
"[52]\tvalid_0's ndcg@1: 0.52419\tvalid_0's ndcg@2: 0.562064\tvalid_0's ndcg@3: 0.607791\tvalid_0's ndcg@4: 0.622855\tvalid_0's ndcg@5: 0.633746\n", | |
"[53]\tvalid_0's ndcg@1: 0.548952\tvalid_0's ndcg@2: 0.568401\tvalid_0's ndcg@3: 0.608267\tvalid_0's ndcg@4: 0.625283\tvalid_0's ndcg@5: 0.63548\n", | |
"[54]\tvalid_0's ndcg@1: 0.537524\tvalid_0's ndcg@2: 0.565081\tvalid_0's ndcg@3: 0.605426\tvalid_0's ndcg@4: 0.621282\tvalid_0's ndcg@5: 0.633045\n", | |
"[55]\tvalid_0's ndcg@1: 0.548952\tvalid_0's ndcg@2: 0.568401\tvalid_0's ndcg@3: 0.606413\tvalid_0's ndcg@4: 0.62357\tvalid_0's ndcg@5: 0.633424\n", | |
"[56]\tvalid_0's ndcg@1: 0.548952\tvalid_0's ndcg@2: 0.568401\tvalid_0's ndcg@3: 0.600281\tvalid_0's ndcg@4: 0.621657\tvalid_0's ndcg@5: 0.631268\n", | |
"[57]\tvalid_0's ndcg@1: 0.535619\tvalid_0's ndcg@2: 0.559698\tvalid_0's ndcg@3: 0.59927\tvalid_0's ndcg@4: 0.621441\tvalid_0's ndcg@5: 0.631479\n", | |
"[58]\tvalid_0's ndcg@1: 0.529905\tvalid_0's ndcg@2: 0.563724\tvalid_0's ndcg@3: 0.600262\tvalid_0's ndcg@4: 0.61555\tvalid_0's ndcg@5: 0.641651\n", | |
"[59]\tvalid_0's ndcg@1: 0.549905\tvalid_0's ndcg@2: 0.573926\tvalid_0's ndcg@3: 0.608217\tvalid_0's ndcg@4: 0.624724\tvalid_0's ndcg@5: 0.639956\n", | |
"[60]\tvalid_0's ndcg@1: 0.555619\tvalid_0's ndcg@2: 0.567699\tvalid_0's ndcg@3: 0.614928\tvalid_0's ndcg@4: 0.62234\tvalid_0's ndcg@5: 0.640292\n", | |
"[61]\tvalid_0's ndcg@1: 0.555619\tvalid_0's ndcg@2: 0.56512\tvalid_0's ndcg@3: 0.619991\tvalid_0's ndcg@4: 0.623841\tvalid_0's ndcg@5: 0.643278\n", | |
"[62]\tvalid_0's ndcg@1: 0.535619\tvalid_0's ndcg@2: 0.560594\tvalid_0's ndcg@3: 0.617178\tvalid_0's ndcg@4: 0.618333\tvalid_0's ndcg@5: 0.639012\n", | |
"[63]\tvalid_0's ndcg@1: 0.535619\tvalid_0's ndcg@2: 0.565894\tvalid_0's ndcg@3: 0.618594\tvalid_0's ndcg@4: 0.62041\tvalid_0's ndcg@5: 0.639891\n", | |
"[64]\tvalid_0's ndcg@1: 0.555619\tvalid_0's ndcg@2: 0.57042\tvalid_0's ndcg@3: 0.617583\tvalid_0's ndcg@4: 0.623142\tvalid_0's ndcg@5: 0.645707\n", | |
"[65]\tvalid_0's ndcg@1: 0.555619\tvalid_0's ndcg@2: 0.575319\tvalid_0's ndcg@3: 0.619513\tvalid_0's ndcg@4: 0.624902\tvalid_0's ndcg@5: 0.648107\n", | |
"[66]\tvalid_0's ndcg@1: 0.555619\tvalid_0's ndcg@2: 0.575319\tvalid_0's ndcg@3: 0.62022\tvalid_0's ndcg@4: 0.622008\tvalid_0's ndcg@5: 0.641083\n", | |
"[67]\tvalid_0's ndcg@1: 0.555619\tvalid_0's ndcg@2: 0.575319\tvalid_0's ndcg@3: 0.618433\tvalid_0's ndcg@4: 0.627618\tvalid_0's ndcg@5: 0.641823\n", | |
"[68]\tvalid_0's ndcg@1: 0.555619\tvalid_0's ndcg@2: 0.580995\tvalid_0's ndcg@3: 0.621228\tvalid_0's ndcg@4: 0.627007\tvalid_0's ndcg@5: 0.648201\n", | |
"[69]\tvalid_0's ndcg@1: 0.555619\tvalid_0's ndcg@2: 0.577898\tvalid_0's ndcg@3: 0.617719\tvalid_0's ndcg@4: 0.63029\tvalid_0's ndcg@5: 0.654197\n", | |
"[70]\tvalid_0's ndcg@1: 0.562286\tvalid_0's ndcg@2: 0.586041\tvalid_0's ndcg@3: 0.622397\tvalid_0's ndcg@4: 0.634144\tvalid_0's ndcg@5: 0.658216\n", | |
"[71]\tvalid_0's ndcg@1: 0.562286\tvalid_0's ndcg@2: 0.58864\tvalid_0's ndcg@3: 0.623986\tvalid_0's ndcg@4: 0.63175\tvalid_0's ndcg@5: 0.655262\n", | |
"[72]\tvalid_0's ndcg@1: 0.555619\tvalid_0's ndcg@2: 0.586608\tvalid_0's ndcg@3: 0.628457\tvalid_0's ndcg@4: 0.627486\tvalid_0's ndcg@5: 0.652192\n", | |
"[73]\tvalid_0's ndcg@1: 0.555619\tvalid_0's ndcg@2: 0.586608\tvalid_0's ndcg@3: 0.620345\tvalid_0's ndcg@4: 0.636233\tvalid_0's ndcg@5: 0.648029\n", | |
"[74]\tvalid_0's ndcg@1: 0.568952\tvalid_0's ndcg@2: 0.585725\tvalid_0's ndcg@3: 0.623823\tvalid_0's ndcg@4: 0.641507\tvalid_0's ndcg@5: 0.646475\n", | |
"[75]\tvalid_0's ndcg@1: 0.568952\tvalid_0's ndcg@2: 0.589191\tvalid_0's ndcg@3: 0.623518\tvalid_0's ndcg@4: 0.63966\tvalid_0's ndcg@5: 0.653877\n", | |
"[76]\tvalid_0's ndcg@1: 0.568952\tvalid_0's ndcg@2: 0.578357\tvalid_0's ndcg@3: 0.619524\tvalid_0's ndcg@4: 0.635502\tvalid_0's ndcg@5: 0.649334\n", | |
"[77]\tvalid_0's ndcg@1: 0.568952\tvalid_0's ndcg@2: 0.567899\tvalid_0's ndcg@3: 0.614403\tvalid_0's ndcg@4: 0.633194\tvalid_0's ndcg@5: 0.645866\n", | |
"[78]\tvalid_0's ndcg@1: 0.575619\tvalid_0's ndcg@2: 0.58039\tvalid_0's ndcg@3: 0.618183\tvalid_0's ndcg@4: 0.63676\tvalid_0's ndcg@5: 0.651016\n", | |
"[79]\tvalid_0's ndcg@1: 0.575619\tvalid_0's ndcg@2: 0.569408\tvalid_0's ndcg@3: 0.611396\tvalid_0's ndcg@4: 0.637912\tvalid_0's ndcg@5: 0.650479\n", | |
"[80]\tvalid_0's ndcg@1: 0.559619\tvalid_0's ndcg@2: 0.562312\tvalid_0's ndcg@3: 0.607836\tvalid_0's ndcg@4: 0.633675\tvalid_0's ndcg@5: 0.645219\n", | |
"[81]\tvalid_0's ndcg@1: 0.559619\tvalid_0's ndcg@2: 0.573645\tvalid_0's ndcg@3: 0.613135\tvalid_0's ndcg@4: 0.642369\tvalid_0's ndcg@5: 0.646426\n", | |
"[82]\tvalid_0's ndcg@1: 0.582286\tvalid_0's ndcg@2: 0.579299\tvalid_0's ndcg@3: 0.614961\tvalid_0's ndcg@4: 0.639335\tvalid_0's ndcg@5: 0.648889\n", | |
"[83]\tvalid_0's ndcg@1: 0.582286\tvalid_0's ndcg@2: 0.583198\tvalid_0's ndcg@3: 0.621661\tvalid_0's ndcg@4: 0.650219\tvalid_0's ndcg@5: 0.650972\n", | |
"[84]\tvalid_0's ndcg@1: 0.565143\tvalid_0's ndcg@2: 0.574319\tvalid_0's ndcg@3: 0.59934\tvalid_0's ndcg@4: 0.626438\tvalid_0's ndcg@5: 0.64397\n", | |
"[85]\tvalid_0's ndcg@1: 0.575619\tvalid_0's ndcg@2: 0.57291\tvalid_0's ndcg@3: 0.599443\tvalid_0's ndcg@4: 0.631623\tvalid_0's ndcg@5: 0.642651\n", | |
"[86]\tvalid_0's ndcg@1: 0.558476\tvalid_0's ndcg@2: 0.56793\tvalid_0's ndcg@3: 0.591332\tvalid_0's ndcg@4: 0.629275\tvalid_0's ndcg@5: 0.64139\n", | |
"[87]\tvalid_0's ndcg@1: 0.558476\tvalid_0's ndcg@2: 0.56793\tvalid_0's ndcg@3: 0.597142\tvalid_0's ndcg@4: 0.631583\tvalid_0's ndcg@5: 0.643919\n", | |
"[88]\tvalid_0's ndcg@1: 0.575619\tvalid_0's ndcg@2: 0.57291\tvalid_0's ndcg@3: 0.597635\tvalid_0's ndcg@4: 0.632973\tvalid_0's ndcg@5: 0.64215\n", | |
"[89]\tvalid_0's ndcg@1: 0.575619\tvalid_0's ndcg@2: 0.57291\tvalid_0's ndcg@3: 0.604897\tvalid_0's ndcg@4: 0.631643\tvalid_0's ndcg@5: 0.643375\n", | |
"[90]\tvalid_0's ndcg@1: 0.575619\tvalid_0's ndcg@2: 0.574972\tvalid_0's ndcg@3: 0.591966\tvalid_0's ndcg@4: 0.62261\tvalid_0's ndcg@5: 0.643703\n", | |
"[91]\tvalid_0's ndcg@1: 0.575619\tvalid_0's ndcg@2: 0.574972\tvalid_0's ndcg@3: 0.601625\tvalid_0's ndcg@4: 0.615755\tvalid_0's ndcg@5: 0.635512\n", | |
"[92]\tvalid_0's ndcg@1: 0.562286\tvalid_0's ndcg@2: 0.566278\tvalid_0's ndcg@3: 0.598308\tvalid_0's ndcg@4: 0.618188\tvalid_0's ndcg@5: 0.633606\n", | |
"[93]\tvalid_0's ndcg@1: 0.562286\tvalid_0's ndcg@2: 0.56112\tvalid_0's ndcg@3: 0.604827\tvalid_0's ndcg@4: 0.627074\tvalid_0's ndcg@5: 0.637866\n", | |
"[94]\tvalid_0's ndcg@1: 0.562286\tvalid_0's ndcg@2: 0.56112\tvalid_0's ndcg@3: 0.606963\tvalid_0's ndcg@4: 0.628069\tvalid_0's ndcg@5: 0.641482\n", | |
"[95]\tvalid_0's ndcg@1: 0.562286\tvalid_0's ndcg@2: 0.56112\tvalid_0's ndcg@3: 0.603114\tvalid_0's ndcg@4: 0.636515\tvalid_0's ndcg@5: 0.640139\n", | |
"[96]\tvalid_0's ndcg@1: 0.562286\tvalid_0's ndcg@2: 0.56112\tvalid_0's ndcg@3: 0.604542\tvalid_0's ndcg@4: 0.630714\tvalid_0's ndcg@5: 0.637942\n", | |
"[97]\tvalid_0's ndcg@1: 0.575619\tvalid_0's ndcg@2: 0.566348\tvalid_0's ndcg@3: 0.600672\tvalid_0's ndcg@4: 0.637343\tvalid_0's ndcg@5: 0.642449\n", | |
"[98]\tvalid_0's ndcg@1: 0.575619\tvalid_0's ndcg@2: 0.566348\tvalid_0's ndcg@3: 0.600672\tvalid_0's ndcg@4: 0.63211\tvalid_0's ndcg@5: 0.641266\n", | |
"[99]\tvalid_0's ndcg@1: 0.545143\tvalid_0's ndcg@2: 0.564027\tvalid_0's ndcg@3: 0.595109\tvalid_0's ndcg@4: 0.631733\tvalid_0's ndcg@5: 0.64311\n", | |
"[100]\tvalid_0's ndcg@1: 0.575619\tvalid_0's ndcg@2: 0.572542\tvalid_0's ndcg@3: 0.608138\tvalid_0's ndcg@4: 0.641857\tvalid_0's ndcg@5: 0.646253\n" | |
] | |
} | |
], | |
"source": [ | |
"model = lgb.train(train_params, train_dataset, valid_sets=[test_dataset])" | |
] | |
}, | |
{ | |
"cell_type": "code", | |
"execution_count": 18, | |
"metadata": {}, | |
"outputs": [ | |
{ | |
"data": { | |
"text/plain": [ | |
"'2.2.3'" | |
] | |
}, | |
"execution_count": 18, | |
"metadata": {}, | |
"output_type": "execute_result" | |
} | |
], | |
"source": [ | |
"lgb.__version__" | |
] | |
}, | |
{ | |
"cell_type": "code", | |
"execution_count": null, | |
"metadata": {}, | |
"outputs": [], | |
"source": [] | |
} | |
], | |
"metadata": { | |
"kernelspec": { | |
"display_name": "Python 3", | |
"language": "python", | |
"name": "python3" | |
}, | |
"language_info": { | |
"codemirror_mode": { | |
"name": "ipython", | |
"version": 3 | |
}, | |
"file_extension": ".py", | |
"mimetype": "text/x-python", | |
"name": "python", | |
"nbconvert_exporter": "python", | |
"pygments_lexer": "ipython3", | |
"version": "3.7.2" | |
} | |
}, | |
"nbformat": 4, | |
"nbformat_minor": 2 | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
CLI