提交 03462fc9 编写于 作者: S ShusenTang

🔨 fix bug #97

上级 c9fd01bc
......@@ -131,7 +131,7 @@ def log_rmse(net, features, labels):
with torch.no_grad():
# 将小于1的值设成1,使得取对数时数值更稳定
clipped_preds = torch.max(net(features), torch.tensor(1.0))
rmse = torch.sqrt(2 * loss(clipped_preds.log(), labels.log()).mean())
rmse = torch.sqrt(loss(clipped_preds.log(), labels.log()))
return rmse.item()
```
......@@ -203,12 +203,12 @@ def k_fold(k, X_train, y_train, num_epochs,
```
输出:
```
fold 0, train rmse 0.241054, valid rmse 0.221462
fold 1, train rmse 0.229857, valid rmse 0.268489
fold 2, train rmse 0.231413, valid rmse 0.238157
fold 3, train rmse 0.237733, valid rmse 0.218747
fold 4, train rmse 0.230720, valid rmse 0.258712
5-fold validation: avg train rmse 0.234155, avg valid rmse 0.241113
fold 0, train rmse 0.170585, valid rmse 0.156860
fold 1, train rmse 0.162552, valid rmse 0.190944
fold 2, train rmse 0.164199, valid rmse 0.168767
fold 3, train rmse 0.168698, valid rmse 0.154873
fold 4, train rmse 0.163213, valid rmse 0.183080
5-fold validation: avg train rmse 0.165849, avg valid rmse 0.170905
```
<img width="400" src="../img/chapter03/3.16_output2.png"/>
......@@ -250,7 +250,7 @@ train_and_pred(train_features, test_features, train_labels, test_data, num_epoch
```
输出:
```
train rmse 0.229943
train rmse 0.162085
```
<img width="400" src="../img/chapter03/3.16_output3.png"/>
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册