latihan 4 - decision tree.ipynb

{"nbformat":4,"nbformat_minor":0,"metadata":{"colab":{"name":"latihan 4 - decision tree.ipynb","provenance":[],"collapsed_sections":[],"authorship_tag":"ABX9TyPUl2LBsuX7UdmDruTlmyCW"},"kernelspec":{"name":"python3","display_name":"Python 3"}},"cells":[{"cell_type":"code","metadata":{"id":"apz3nLF2ELGp","colab_type":"code","colab":{}},"source":["from sklearn.tree import DecisionTreeClassifier\n","from sklearn.datasets import load_iris\n","import pandas as pd"],"execution_count":0,"outputs":[]},{"cell_type":"code","metadata":{"id":"oFh2k7nuEidK","colab_type":"code","colab":{"base_uri":"https://localhost:8080/","height":206},"outputId":"4b691b59-c0de-4a1a-eed6-0e81e0303d11","executionInfo":{"status":"ok","timestamp":1588350868626,"user_tz":0,"elapsed":2985,"user":{"displayName":"Agung Kurniawan","photoUrl":"https://lh3.googleusercontent.com/a-/AOh14Ghrb76eETY0EEb-xiAoS5M2HEAneMGxLUO0pzkUoA=s64","userId":"17681440155891072045"}}},"source":["iris = pd.read_csv('Iris.csv')\n","iris.drop('Id', axis=1,inplace=True)\n","iris.head()"],"execution_count":5,"outputs":[{"output_type":"execute_result","data":{"text/html":["<div>\n","<style scoped>\n","    .dataframe tbody tr th:only-of-type {\n","        vertical-align: middle;\n","    }\n","\n","    .dataframe tbody tr th {\n","        vertical-align: top;\n","    }\n","\n","    .dataframe thead th {\n","        text-align: right;\n","    }\n","</style>\n","<table border=\"1\" class=\"dataframe\">\n","  <thead>\n","    <tr style=\"text-align: right;\">\n","      <th></th>\n","      <th>SepalLengthCm</th>\n","      <th>SepalWidthCm</th>\n","      <th>PetalLengthCm</th>\n","      <th>PetalWidthCm</th>\n","      <th>Species</th>\n","    </tr>\n","  </thead>\n","  <tbody>\n","    <tr>\n","      <th>0</th>\n","      <td>5.1</td>\n","      <td>3.5</td>\n","      <td>1.4</td>\n","      <td>0.2</td>\n","      <td>Iris-setosa</td>\n","    </tr>\n","    <tr>\n","      <th>1</th>\n","      <td>4.9</td>\n","      <td>3.0</td>\n","      <td>1.4</td>\n","      <td>0.2</td>\n","      <td>Iris-setosa</td>\n","    </tr>\n","    <tr>\n","      <th>2</th>\n","      <td>4.7</td>\n","      <td>3.2</td>\n","      <td>1.3</td>\n","      <td>0.2</td>\n","      <td>Iris-setosa</td>\n","    </tr>\n","    <tr>\n","      <th>3</th>\n","      <td>4.6</td>\n","      <td>3.1</td>\n","      <td>1.5</td>\n","      <td>0.2</td>\n","      <td>Iris-setosa</td>\n","    </tr>\n","    <tr>\n","      <th>4</th>\n","      <td>5.0</td>\n","      <td>3.6</td>\n","      <td>1.4</td>\n","      <td>0.2</td>\n","      <td>Iris-setosa</td>\n","    </tr>\n","  </tbody>\n","</table>\n","</div>"],"text/plain":["   SepalLengthCm  SepalWidthCm  PetalLengthCm  PetalWidthCm      Species\n","0            5.1           3.5            1.4           0.2  Iris-setosa\n","1            4.9           3.0            1.4           0.2  Iris-setosa\n","2            4.7           3.2            1.3           0.2  Iris-setosa\n","3            4.6           3.1            1.5           0.2  Iris-setosa\n","4            5.0           3.6            1.4           0.2  Iris-setosa"]},"metadata":{"tags":[]},"execution_count":5}]},{"cell_type":"markdown","metadata":{"id":"WzTMBkFBF0wG","colab_type":"text"},"source":["Split attribute & label for training"]},{"cell_type":"code","metadata":{"id":"jHjkdSMDF6Ph","colab_type":"code","colab":{}},"source":["x = iris[['SepalLengthCm', 'SepalWidthCm','PetalLengthCm','PetalWidthCm']]\n","y= iris['Species']"],"execution_count":0,"outputs":[]},{"cell_type":"markdown","metadata":{"id":"cxsMDQygGe6f","colab_type":"text"},"source":["Create Decision Tree model"]},{"cell_type":"code","metadata":{"id":"dbttNeenGjAW","colab_type":"code","colab":{"base_uri":"https://localhost:8080/","height":34},"outputId":"9ef4feb4-15ab-458d-bf66-01679765f996","executionInfo":{"status":"ok","timestamp":1588351693091,"user_tz":0,"elapsed":983,"user":{"displayName":"Agung Kurniawan","photoUrl":"https://lh3.googleusercontent.com/a-/AOh14Ghrb76eETY0EEb-xiAoS5M2HEAneMGxLUO0pzkUoA=s64","userId":"17681440155891072045"}}},"source":["model = DecisionTreeClassifier()\n","model.fit(x,y)\n","\n","model.predict([[6.2,3.4,5.4,2.3]])"],"execution_count":7,"outputs":[{"output_type":"execute_result","data":{"text/plain":["array(['Iris-virginica'], dtype=object)"]},"metadata":{"tags":[]},"execution_count":7}]}]}