paddle练习(四)使用高层API完成模型训练&预测

也可以用飞桨的高层API来做线性回归训练,高层API相较于底层API更加的简洁方便。

# 方式2:使用高层API完成模型训练&预测
import paddle
paddle.set_default_dtype("float64")

# step1:用高层API定义数据集,无需进行数据处理等,高层API为你一条龙搞定
train_dataset = paddle.text.datasets.UCIHousing(mode='train')
eval_dataset = paddle.text.datasets.UCIHousing(mode='test')

# step2:定义模型
class UCIHousing(paddle.nn.Layer):
    def __init__(self):
        super(UCIHousing, self).__init__()
        self.fc = paddle.nn.Linear(13, 1, None)

    def forward(self, input):
        pred = self.fc(input)
        return pred

# step3:训练模型
model = paddle.Model(UCIHousing())
model.prepare(paddle.optimizer.Adam(parameters=model.parameters()),
              paddle.nn.MSELoss())
model.fit(train_dataset, eval_dataset, epochs=5, batch_size=8, verbose=1)

运行结果

Cache file C:\Users\YANG\.cache\paddle\dataset\uci_housing\housing.data not found, downloading http://paddlemodels.bj.bcebos.com/uci_housing/housing.data 
Begin to download
item 12/12 [==========================>...] - ETA: 0s - 5ms/itemThe loss value printed in the log is the current step, and the metric is the average value of previous steps.

Download finished
Epoch 1/5
step 51/51 [==============================] - loss: 617.5771 - 1ms/step          
Eval begin...
step 13/13 [==============================] - loss: 381.6220 - 780us/step          
Eval samples: 102
Epoch 2/5
step 51/51 [==============================] - loss: 414.8060 - 1ms/step          
Eval begin...
step 13/13 [==============================] - loss: 379.1168 - 849us/step          
Eval samples: 102
Epoch 3/5
step 51/51 [==============================] - loss: 415.8149 - 1ms/step          
Eval begin...
step 13/13 [==============================] - loss: 376.6546 - 812us/step          
Eval samples: 102
Epoch 4/5
step 51/51 [==============================] - loss: 431.1117 - 1ms/step          
Eval begin...
step 13/13 [==============================] - loss: 374.2060 - 828us/step          
Eval samples: 102
Epoch 5/5
step 51/51 [==============================] - loss: 458.0943 - 1ms/step          
Eval begin...
step 13/13 [==============================] - loss: 371.7430 - 724us/step          
Eval samples: 102

Process finished with exit code 0

可以看出,非常简便,自动下载house.data

无需你自己做数据预处理

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值