使用用户输入测试预测模型

我是ML的初学者,但是我正在做一个大学项目,并且我能够成功训练模型,但是我不确定如何测试用户输入。我的项目是检查为一个人输入的数据是否为糖尿病。

数据CSV:

Pregnancies Glucose BloodPressure Skinthickness Insulin BMI DiabetesPedigreeFunction Age Outcome
6   148 72  35  0   33.6    0.627   50  1
1   85  66  29  0   26.6    0.351   31  0
8   183 64  0   0   23.3    0.672   32  1
1   89  66  23  94  28.1    0.167   21  0
0   137 40  35  168 43.1    2.288   33  1
5   116 74  0   0   25.6    0.201   30  0
3   78  50  32  88  31  0.248   26  1
10  115 0   0   0   35.3    0.134   29  0
2   197 70  45  543 30.5    0.158   53  1

使用用户输入测试预测模型

代码:

from sklearn.ensemble import RandomForestClassifier
random_forest_model = RandomForestClassifier(random_state=10)
random_forest_model.fit(X_train,y_train.ravel())

predict_train_data = random_forest_model.predict(X_test)
from sklearn import metrics
print("accuracy = {0:.3f}".format(metrics.accuracy_score(y_test,predict_train_data)))

用户输入代码:

print("Enter your own data to test the model:")
pregnancy = int(input("Enter Pregnancy:"))
glucose = int(input("Enter Glucose:"))
bloodpressure = int(input("Enter Blood Pressue:"))
skinthickness = int(input("Enter Skin Thickness:"))
insulin = int(input("Enter Insulin:"))
bmi = float(input("Enter BMI:"))
DiabetesPedigreeFunction = float(input("Enter DiabetesPedigreeFunction:"))
age = int(input("Enter Age:"))
userInput = [pregnancy,glucose,bloodpressure,skinthickness,insulin,bmi,DiabetesPedigreeFunction,age]

我希望它返回1-如果是糖尿病,则返回0-如果是非糖尿病


编辑-添加了x_train和y_train:

from sklearn.model_selection import train_test_split
feature_columns = ['Pregnancies','Glucose','BloodPressure','Skinthickness','Insulin','BMI','DiabetesPedigreeFunction','Age']
predicted_class = ['Outcome']

X = data[feature_columns].values
y = data[predicted_class].values

X_train,X_test,y_train,y_test = train_test_split(X,y,test_size = 0.30,random_state=10)

from sklearn.ensemble import RandomForestClassifier
random_forest_model = RandomForestClassifier(random_state=10)
random_forest_model.fit(X_train,y_train.ravel())
xiaobestavril 回答:使用用户输入测试预测模型

尝试

result = random_forest_model.predict([user_input])[0]

因为模型需要多个输入(二维数组)并返回每个元素的预测(观测值列表)。

本文链接:https://www.f2er.com/3159055.html

大家都在问