diff --git a/decision_tree/dt_author_id.py b/decision_tree/dt_author_id.py index 006afb8357b..10fceed74bb 100644 --- a/decision_tree/dt_author_id.py +++ b/decision_tree/dt_author_id.py @@ -24,7 +24,13 @@ ######################################################### ### your code goes here ### +from sklearn import tree +clf=tree.DecisionTreeClassifier() +clf.fit(features_train, labels_train ) +pred=clf.predict(features_test) +from sklearn.metrics import accuracy_score +acc=accuracy_score(pred,labels_test) #########################################################