{"cells":[
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": "import pandas as pd\nfrom sklearn import tree\n\ntest = pd.read_csv('../input/test.csv')\ntrain = pd.read_csv('../input/train.csv')\n\nclf = tree.DecisionTreeClassifier()\n\ntrainLabel = train.label\ntrain = train.drop('label',1)\n\nclf.fit(train,trainLabel)\n\ndt = clf.predict(test)"
 },
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": "predictions = pd.DataFrame(data=dt,columns=[\"label\"])\npredictions.ImageId = list(range(1,len(test)+1))"
 },
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": "predictions.to_csv(\"sklearn_decisiontree.csv\",index_label=\"ImageId\", sep='\\t')"
 },
 {
  "cell_type": "code",
  "execution_count": null,
  "metadata": {
   "collapsed": false
  },
  "outputs": [],
  "source": ""
 }
],"metadata":{"kernelspec":{"display_name":"Python 3","language":"python","name":"python3"}}, "nbformat": 4, "nbformat_minor": 0}