2018.1.7

最新推荐文章于 2024-09-05 08:25:34 发布

swy_swy_swy

最新推荐文章于 2024-09-05 08:25:34 发布

阅读量163

点赞数 1

分类专栏：琉璃神社

本文链接：https://blog.csdn.net/swy_swy_swy/article/details/105770988

版权

琉璃神社专栏收录该内容

907 篇文章 874 订阅

订阅专栏

2018.1.7

import numpy as np
from sklearn import tree
from sklearn.model_selection import cross_val_score

first_set = np.loadtxt(
    "../datas/train_set.csv", str, delimiter=",", usecols=(6, 12), skiprows=1
)  # balance & duration
first_set_label = np.loadtxt(
    "../datas/train_set.csv", str, delimiter=",", usecols=17, skiprows=1
)  # 1st labels
# print(first_set)
# decision tree begins
clf = tree.DecisionTreeClassifier()
clf = clf.fit(first_set, first_set_label)
scores = cross_val_score(clf, first_set, first_set_label, cv=10)
print(scores)