fix bug in ranking tutorial ( )
"\n",
"train_with_weights = Pool(\n",
" data=X_train,\n",
- " label=y_train,\n",
- " group_id=queries_train\n",
+ " label=best_docs_train,\n",
+ " group_id=queries_train,\n",
+ " group_weight=create_weights(queries_train)\n",
")\n",
"\n",
"test_with_weights = Pool(\n",
" data=X_test,\n",
- " label=y_test,\n",
- " group_id=queries_test\n",
+ " label=best_docs_test,\n",
+ " group_id=queries_test,\n",
+ " group_weight=create_weights(queries_test)\n",
")\n",
"\n",
"fit_model(\n",
