【发布时间】:2019-07-19 22:08:13
【问题描述】:
我写了一个函数,它接受数据集(excel / pandas)和一些值,然后用决策树分类器预测结果。我已经用 sklearn 做到了。 你能帮我解决这个问题吗,我浏览了网络和这个网站,但我找不到有效的答案。 我试过这样做,但它不起作用:
from sklearn.metrics import accuracy_score
score = accuracy_score(variable_list, result_list)
这是我得到的错误:
ValueError: Classification metrics can't handle a mix of continuous-multioutput and multiclass targets
这是代码(为了准确起见,我删除了代码)
import pandas as pd
import math
import xlrd
from sklearn.model_selection import train_test_split
from sklearn import tree
def predict_concrete_class(input_data, cement, blast_fur_slug,fly_ash,
water, superpl, coarse_aggr, fine_aggr, days):
data_for_tree = concrete_strenght_class(input_data)
variable_list = []
result_list = []
for index, row in data_for_tree.iterrows():
variable = row.tolist()
variable = variable[0:8]
variable_list.append(variable)
result_list.append(row[-1])
decision_tree = tree.DecisionTreeClassifier()
decision_tree = decision_tree.fit(variable_list,result_list)
input_values = [cement, blast_fur_slug, fly_ash, water, superpl, coarse_aggr, fine_aggr, days]
prediction = decision_tree.predict([input_values])
info = "Prediction of future concrete class after "+ str(days)+" days: "+ str(prediction[0])
return info
print(predict_concrete_class(data, 500, 0, 0, 200, 0, 1125, 613, 3))
【问题讨论】:
-
您到底在哪里遇到问题?
-
我不知道该怎么做。我试图在测试和训练中拆分值,但我不知道该怎么做,以及我应该使用什么值进行测试和训练
-
@Chris 我也更新了我的问题,请检查一下
-
您向我们提供了您尝试过的代码,但您没有告诉我们出了什么问题。您收到错误代码吗?如果有,是哪一个?
-
这是错误:[ValueError:分类指标无法处理连续多输出和多类目标的混合]
标签: python scikit-learn classification decision-tree