{"cells":[
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": "import numpy as np\nimport scipy as sp\nimport pandas as pd\nimport matplotlib.pyplot as plt\nimport seaborn as sns\nfrom sklearn.ensemble import RandomForestClassifier\nfrom sklearn.feature_selection import RFE\n\ntrain_df = pd.read_csv(\"../input/train.csv\",nrows=2000)\nX = train_df.drop(\"label\",axis=1)\nprint (X.values.shape)\ny = train_df.label\n\nclf = RandomForestClassifier(n_estimators=60,max_depth=5)\nrfe = RFE(estimator=clf)\nrfe.fit(X,y)\nrank = rfe.ranking_.reshape((28,28))\nsns.heatmap(rank)\nplt.title(\"Ranking of pixels with RFE\")\nplt.show()"
 }
],"metadata":{"kernelspec":{"display_name":"Python 3","language":"python","name":"python3"}}, "nbformat": 4, "nbformat_minor": 0}