{
"cells": [
{
"cell_type": "code",
"execution_count": 6,
"metadata": {},
"outputs": [],
"source": [
"import numpy as np\n",
"import pandas as pd"
]
},
{
"cell_type": "code",
"execution_count": 20,
"metadata": {},
"outputs": [],
"source": [
"#分類問題\n",
"df_past = pd.read_csv(\"4-6_user_data.csv\")\n",
"df_future = pd.read_csv(\"4-6_user_data_future.csv\")\n",
"\n",
"X_name = [\"sex\",\"student\",\"stay time\"]\n",
"y_name = \"registration\"\n",
"\n",
"X = df_past[X_name]\n",
"y = df_past[y_name]"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"#回帰問題\n",
"df_past = pd.read_csv(\"4-6_sales.csv\")\n",
"df_future = pd.read_csv(\"4-6_sales_future.csv\")\n",
"\n",
"X_name = [\"temperature\", \"price\", \"rainy\"]\n",
"y_name = \"sales\"\n",
"\n",
"X = df_past[X_name]\n",
"y = df_past[y_name]"
]
},
{
"cell_type": "code",
"execution_count": 13,
"metadata": {},
"outputs": [],
"source": [
"from sklearn import tree\n",
"from dtreeviz.trees import *\n",
"import graphviz"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"dtree = tree.DecisionTreeClassifier(max_depth=2)\n",
"dtree.fit(X,y)\n",
"\n",
"viz = dtreeviz(dtree,X,y,\n",
" target_name = y_name,\n",
" feature_names = X_name,\n",
" class_names = [\"not register\",\"register\"],\n",
" X = [1,1,50]\n",
" )\n",
"\n",
"\n",
"viz"
]
},
{
"cell_type": "code",
"execution_count": 23,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"array([1, 0, 0, 1, 1, 0, 0, 1, 0, 1])"
]
},
"execution_count": 23,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"dtree.predict(df_future)"
]
},
{
"cell_type": "code",
"execution_count": 26,
"metadata": {},
"outputs": [
{
"data": {
"image/svg+xml": [
""
],
"text/plain": [
""
]
},
"execution_count": 26,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"dtree = tree.DecisionTreeRegressor(max_depth=2)\n",
"dtree.fit(X,y)\n",
"\n",
"viz = dtreeviz(dtree,X,y,\n",
" target_name = y_name,\n",
" feature_names = X_name,\n",
" X = [20,220,0]\n",
" )\n",
"\n",
"\n",
"viz"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.7.4"
}
},
"nbformat": 4,
"nbformat_minor": 2
}