【课程作业经验】基于MindSpore疫苗接种数据预测

基于mindspore实现疫苗接种数据预测

基于机器学习实践课程完成的相关使用mindspore深度学习框架完成的任务，写一些分享心得。

数据导入与准备

数据下载链接：百度网盘 请输入提取码 提取码：23vb

疫苗接种数据集包含1983年-2016年疫苗接种数据，其形式如下图所示：

读取数据并进行训练集、测试集分割：

df = pd.read_csv("vaccine.csv")
features = df["Year"]
target = df["Values"]
split_num = int(len(features)*0.7)

X_train = features[:split_num]
y_train = target[:split_num]    

X_test = features[split_num:]
y_test = target[split_num:]

# 创建数据
X_train,X_test = poly_transform(X_train,X_test,degree=1)
ds_train=create_dataset(X_train,y_train,batch_number,repeat_number)
复制

构建多项式特征：

def poly_transform(X_train,X_test,degree=2):
    '''
    多项式特征
    '''
    poly_features_2 = PolynomialFeatures(degree = degree, include_bias=False)
    poly_X_train_2 = poly_features_2.fit_transform(X_train.values.reshape(len(X_train),1))
    poly_X_test_2 = poly_features_2.fit_transform(X_test.values.reshape(len(X_test),1))

    return poly_X_train_2,poly_X_test_2
复制

全连接网络模型建立

构建一层1->1的网络，比较简单：

class LinearNet(nn.Cell):
    def __init__(self,n=1):
        super(LinearNet,self).__init__()
        # 定义一个线形层，同时初始