log_rmse()是一个泛用的损失函数，其定义不应该只接受train_features，而且事实上在后面`def train()`函数里也调用了`log_rmse(net, test_features, test_labels)`