[catboost] fix bug in model_export_as_json_tutorial

ref:a2af4336973913f3264c57870a362fafc3394b03

Author: ek-ak

model_analysis/model_export_as_json_tutorial.ipynb

@@ -31,7 +31,6 @@
  "source": [
  "import catboost\n",
  "from catboost import datasets\n",
- "from catboost.utils import create_cd\n",
  "import os\n",
  "import numpy as np\n",
  "\n",
@@ -40,7 +39,7 @@
  "pool = catboost.Pool(\n",
  " data=X[:1000], # top 1000 documents are enough for this example\n",
  " label=Y[:1000],\n",
- " feature_names=dict(enumerate(X))\n",
+ " feature_names=list(X.columns)\n",
  ")"
  ]
  },
@@ -130,19 +129,20 @@
  "name": "stdout",
  "output_type": "stream",
  "text": [
- "Git info:\n",
- " Commit: 573374100b7b767846cd30e914cece73cfec17a5\n",
- " Author: 'format:Stanislav Kirillov <staskirillov@gmail.com>'\n",
- "\n",
- " Summary: version v0.10.3 ref:a691b211b401900433abee0a8972132e97b4cbec\n",
+ "Svn info:\n",
+ " URL: svn+ssh://arcadia.yandex.ru/arc/trunk/arcadia\n",
+ " Last Changed Rev: 5437739\n",
+ " Last Changed Author: dkvasov\n",
+ " Last Changed Date: 2019-08-08T13:43:02.471284Z\n",
  "\n",
  "Other info:\n",
- " Build by: go\n",
- " Top src dir: /var/lib/go-agent/pipelines/BuildMaster/catboost.git\n",
- " Top build dir: /nvme/.ya/build/build_root/ap4q/000eb9\n",
- " Hostname: ip-172-31-39-161\n",
+ " Build by: eermishkina\n",
+ " Top src dir: /place/home/eermishkina/trunc/arcadia\n",
+ " Top build dir: /home/eermishkina/.ya/build\n",
+ " Hostname: su57.search.yandex.net\n",
  " Host information: \n",
- " Linux ip-172-31-39-161 4.4.0-1065-aws #75-Ubuntu SMP Fri Aug 10 11:14:32 UTC 2018 x86_64 x86_64 x86_64 GNU/Linux\n",
+ " Linux su57.search.yandex.net 4.4.88-42 #1 SMP Mon Sep 18 14:33:37 UTC 2017 x86_64\n",
+ "\n",
  " \n"
  ]
  }
@@ -174,7 +174,7 @@
  {
  "data": {
  "text/plain": [
- "[u'cat_features_hash', u'ctrs', u'float_features', u'categorical_features']"
+ "[u'float_features']"
  ]
  },
  "execution_count": 6,
@@ -221,7 +221,7 @@
  {
  "data": {
  "text/plain": [
- "{u'borders': [8.5],\n",
+ "{u'borders': [83.5],\n",
  " u'feature_index': 0,\n",
  " u'flat_feature_index': 0,\n",
  " u'has_nans': False,\n",
@@ -281,30 +281,46 @@
  "text": [
  "{\n",
  " \"leaf_values\": [\n",
- " -0.3465735863, \n",
- " 0, \n",
- " -0.5493061443, \n",
- " 0.4236489302\n",
+ " 0.022173912547853145, \n",
+ " 0.017826086558078085, \n",
+ " 0.011304347573415556, \n",
+ " -0.02565217333967221, \n",
+ " -0.02565217333967221, \n",
+ " 0.07652173742004059, \n",
+ " -0.016956521360122625, \n",
+ " -0.019130434355010134, \n",
+ " -0.02130434734989765, \n",
+ " -0.019130434355010148, \n",
+ " 0.02407969585645712, \n",
+ " 0.027495255552409406, \n",
+ " 0.0035863376807432745, \n",
+ " -0.026869069608165402, \n",
+ " -0.028292219481478875, \n",
+ " 0.07398772840759553, \n",
+ " -0.004233128739738201, \n",
+ " -0.012975459832675437, \n",
+ " -0.0286196312621424, \n",
+ " -0.02815950857304045\n",
  " ], \n",
  " \"splits\": [\n",
  " {\n",
- " \"split_index\": 13, \n",
- " \"float_feature_index\": 88, \n",
- " \"border_id\": 4.976368427, \n",
+ " \"split_index\": 4, \n",
+ " \"float_feature_index\": 16, \n",
+ " \"border\": 10.407758712768555, \n",
  " \"split_type\": \"FloatFeature\"\n",
  " }, \n",
  " {\n",
- " \"split_index\": 1, \n",
- " \"float_feature_index\": 6, \n",
- " \"border_id\": 0.07142850012, \n",
+ " \"split_index\": 3, \n",
+ " \"float_feature_index\": 13, \n",
+ " \"border\": 3.5, \n",
  " \"split_type\": \"FloatFeature\"\n",
  " }\n",
  " ], \n",
  " \"leaf_weights\": [\n",
- " 72, \n",
- " 0, \n",
- " 388, \n",
- " 540\n",
+ " 123, \n",
+ " 54, \n",
+ " 512, \n",
+ " 311\n",
  " ]\n",
  "}\n"
  ]
@@ -356,7 +372,7 @@
  {
  "data": {
  "text/plain": [
- "{u'borders': [8.5],\n",
+ "{u'borders': [83.5],\n",
  " u'feature_index': 0,\n",
  " u'flat_feature_index': 0,\n",
  " u'has_nans': False,\n",
@@ -417,13 +433,13 @@
  {
  "data": {
  "text/plain": [
- "[{u'border_id': 4.976368427,\n",
- " u'float_feature_index': 88,\n",
- " u'split_index': 13,\n",
+ "[{u'border': 10.407758712768555,\n",
+ " u'float_feature_index': 16,\n",
+ " u'split_index': 4,\n",
  " u'split_type': u'FloatFeature'},\n",
- " {u'border_id': 0.07142850012,\n",
- " u'float_feature_index': 6,\n",
- " u'split_index': 1,\n",
+ " {u'border': 3.5,\n",
+ " u'float_feature_index': 13,\n",
+ " u'split_index': 3,\n",
  " u'split_type': u'FloatFeature'}]"
  ]
  },
@@ -445,15 +461,15 @@
  {
  "data": {
  "text/plain": [
- "[{'border_id': 4.976368427,\n",
- " 'flat_feature_index': 88,\n",
- " 'float_feature_index': 88,\n",
- " 'split_index': 13,\n",
+ "[{'border_id': 10.407758712768555,\n",
+ " 'flat_feature_index': 16,\n",
+ " 'float_feature_index': 16,\n",
+ " 'split_index': 4,\n",
  " 'split_type': 'FloatFeature'},\n",
- " {'border_id': 0.07142850012,\n",
- " 'flat_feature_index': 6,\n",
- " 'float_feature_index': 6,\n",
- " 'split_index': 1,\n",
+ " {'border_id': 3.5,\n",
+ " 'flat_feature_index': 13,\n",
+ " 'float_feature_index': 13,\n",
+ " 'split_index': 3,\n",
  " 'split_type': 'FloatFeature'}]"
  ]
  },
@@ -489,26 +505,26 @@
  {
  "data": {
  "text/plain": [
- "{u'leaf_values': [0.04693548282,\n",
- " -0.0212903221,\n",
- " -0.02564516072,\n",
+ "{u'leaf_values': [0.05084745649058943,\n",
+ " -0.025423728245294732,\n",
+ " -0.025423728245294732,\n",
+ " -0.02526315733006127,\n",
+ " 0.04578947266073611,\n",
+ " -0.020526315330674765,\n",
  " 0,\n",
  " 0,\n",
  " 0,\n",
- " -0.02499999944,\n",
- " 0.04499999899,\n",
- " -0.01999999955,\n",
- " -0.02557376992,\n",
- " -0.0181967209,\n",
- " 0.04377049082],\n",
- " u'leaf_weights': [53, 0, 45, 52],\n",
- " u'splits': [{u'border_id': 1.549999952,\n",
+ " -0.025573769920184966,\n",
+ " -0.018196720904746992,\n",
+ " 0.04377049082493207],\n",
+ " u'leaf_weights': [50, 48, 0, 52],\n",
+ " u'splits': [{u'border': 0.800000011920929,\n",
  " u'float_feature_index': 3,\n",
- " u'split_index': 66,\n",
+ " u'split_index': 58,\n",
  " u'split_type': u'FloatFeature'},\n",
- " {u'border_id': 3.400000095,\n",
- " u'float_feature_index': 2,\n",
- " u'split_index': 37,\n",
+ " {u'border': 1.5499999523162842,\n",
+ " u'float_feature_index': 3,\n",
+ " u'split_index': 64,\n",
  " u'split_type': u'FloatFeature'}]}"
  ]
  },
@@ -562,7 +578,9 @@
  {
  "data": {
  "text/plain": [
- "array([-0.96672324, 2.03299365, -0.79820129])"
+ "array([[ 0.46872039, 0.02717889, -0.02942634, -0.23159877, -0.23487417],\n",
+ " [ 0.16342239, 0.23999561, 0.08782089, -0.24248687, -0.24875202],\n",
+ " [ 0.33317769, 0.15813378, -0.0085402 , -0.24050062, -0.24227064]])"
  ]
  },
  "execution_count": 15,
@@ -576,16 +594,18 @@
  },
  {
  "cell_type": "code",
- "execution_count": 21,
+ "execution_count": 16,
  "metadata": {},
  "outputs": [
  {
  "data": {
  "text/plain": [
- "array([-1.09331328, 1.22121605, -0.69495837])"
+ "array([[ 0.27934996, 0.00999965, -0.04288901, -0.12313168, -0.12332892],\n",
+ " [ 0.08037036, 0.13138144, 0.04135186, -0.124843 , -0.12826065],\n",
+ " [ 0.19021373, 0.07576716, -0.01227257, -0.12547529, -0.12823304]])"
  ]
  },
- "execution_count": 21,
+ "execution_count": 16,
  "metadata": {},
  "output_type": "execute_result"
  }
@@ -599,16 +619,18 @@
  },
  {
  "cell_type": "code",
- "execution_count": 20,
+ "execution_count": 17,
  "metadata": {},
  "outputs": [
  {
  "data": {
  "text/plain": [
- "array([ 0.12659004, 0.8117776 , -0.10324292])"
+ "array([[ 0.27934996, 0.00999965, -0.04288901, -0.12313168, -0.12332892],\n",
+ " [ 0.08037036, 0.13138144, 0.04135186, -0.124843 , -0.12826065],\n",
+ " [ 0.19021373, 0.07576716, -0.01227257, -0.12547529, -0.12823304]])"
  ]
  },
- "execution_count": 20,
+ "execution_count": 17,
  "metadata": {},
  "output_type": "execute_result"
  }