python-3.x - 如何在决策树中获取新数据的标签编码器
问题描述
我在以下示例数据的帮助下执行决策树。
所以我把上面的数据转换成LabelEncoder来执行Decision Tree,成功创建了一个DT模型。
所以现在我的要求是我想预测下面的值,那么如何在 python 代码中传递这些值。
为了预测现有值,我可以使用下面的预测代码。
model.predict([[2,1,1]])
完整代码
import pandas as pd
df = pd.read_csv(r"salaries.csv")
df.head()
inputs = df.drop('salary_more_then_100k',axis='columns')
target = df['salary_more_then_100k']
from sklearn.preprocessing import LabelEncoder
le_company = LabelEncoder()
le_job = LabelEncoder()
le_degree = LabelEncoder()
inputs['company_n'] = le_company.fit_transform(inputs['company'])
inputs['job_n'] = le_job.fit_transform(inputs['job'])
inputs['degree_n'] = le_degree.fit_transform(inputs['degree'])
inputs_n = inputs.drop(['company','job','degree'],axis='columns')
from sklearn import tree
model = tree.DecisionTreeClassifier()
model.fit(inputs_n, target)
model.score(inputs_n,target)
解决方案
推荐阅读
- spring-boot - log4j2.xml 的 kafka appender 中的配置更改?
- python - ssl error why installing with pip
- javascript - NgRx-仅当返回先前的服务调用响应时,我应该如何触发服务调用
- java - BiConsumer 使用第一种类型来设置它的字段使用第二种类型
- contact-form-7 - 验证联系表 7 到参与者数据库字段中的电子邮件地址
- django - 在 django 的 models.py 中创建一个列表
- java - 将 Excel 数据发送到 Java 应用程序
- mongodb - MongoDB通过具有不同值的对象数组查找
- jquery - How can I set groups of three list items to be the same height as the tallest?
- ruby-on-rails - Rails - 尝试提供 nginx 缓存图像时出现 404