我有这样一个数据集
如您所见，有一个分类变量是状态
稍后我对分类变量进行编码
如果我想用特定数据测试我的模型，我会这样做
print(regressor.predict([[1,0,1000,2000,3000]]))
效果很好。但我想做的是，在测试时我直接想输入城市名称，比如 New York 或 Florida
我怎样才能做到这一点？
请您参考如下方法:
机器学习模型只能处理数字数据。这就是为什么您必须对“状态”进行编码的原因。 有几种方法可以实现您所说的: a)使用函数返回“状态”的编码值，同时您可以输入类似
print(regressor.predict([[1,0,1000,func("New York"),3000]]))
b) 使用隐式编码，隐式地为每个分类变量创建尽可能多的列。
