本项目的主要目标是建立一个机器学习模型,该模型可以根
据 Covid-19 患者的当前症状、状态和病史预测患者是否处于高
危状态。
给出三个文件:1、训练数据:train_data.csv;2、测试数据:
test_data.csv;3、结果提交样例:test-example.csv。
训练数据给出了一些患者的症状,并标明这种症状是否是高
危病人(USMER 属性)。
测试数据给出了另一些患者的情况,但是并没有告诉你他们
的 USMER 属性,此时需要我们想办法预测出每个患者是否是高
危病人(1 或者 2)。
最终将预测的结果按照顺序整理成结果提交样例的形式,后
面会给出一个软件测试分数。
import pandas as pd
data = pd.read_csv(r'data/train_data.csv')
X = data.drop('USMER', axis=1)
y = data['USMER']
这个代码该怎么写啊