Learn practical skills, build real-world projects, and advance your career
!pip install jovian --upgrade --quiet
!pip install scikit-learn --upgrade
!pip install pandas
Requirement already up-to-date: scikit-learn in /srv/conda/envs/notebook/lib/python3.7/site-packages (0.23.2) Requirement already satisfied, skipping upgrade: threadpoolctl>=2.0.0 in /srv/conda/envs/notebook/lib/python3.7/site-packages (from scikit-learn) (2.1.0) Requirement already satisfied, skipping upgrade: numpy>=1.13.3 in /srv/conda/envs/notebook/lib/python3.7/site-packages (from scikit-learn) (1.19.2) Requirement already satisfied, skipping upgrade: scipy>=0.19.1 in /srv/conda/envs/notebook/lib/python3.7/site-packages (from scikit-learn) (1.5.2) Requirement already satisfied, skipping upgrade: joblib>=0.11 in /srv/conda/envs/notebook/lib/python3.7/site-packages (from scikit-learn) (0.16.0) Requirement already satisfied: pandas in /srv/conda/envs/notebook/lib/python3.7/site-packages (1.1.2) Requirement already satisfied: python-dateutil>=2.7.3 in /srv/conda/envs/notebook/lib/python3.7/site-packages (from pandas) (2.8.1) Requirement already satisfied: pytz>=2017.2 in /srv/conda/envs/notebook/lib/python3.7/site-packages (from pandas) (2020.1) Requirement already satisfied: numpy>=1.15.4 in /srv/conda/envs/notebook/lib/python3.7/site-packages (from pandas) (1.19.2) Requirement already satisfied: six>=1.5 in /srv/conda/envs/notebook/lib/python3.7/site-packages (from python-dateutil>=2.7.3->pandas) (1.15.0)
import jovian
jovian.commit(project='machine-learning-3', environment=None)
[jovian] Attempting to save notebook..
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
from sklearn import tree

music_datas = pd.read_csv('music.csv')
X = music_datas.drop(columns=['genre'])
Y = music_datas['genre']
model = DecisionTreeClassifier()
model.fit(X,Y)

tree.export_graphviz(model, out_file='music-recommender.dot', feature_names=['age','gender'], class_names=sorted(Y.unique()), label='all',rounded=True,filled=True)