fix bugs on house price

all_features contains bool type we need convert all to float32 at first for function d2l.tensor
cout0 · Mar 18, 2024 · 7b71a72 · 7b71a72
1 parent e6b18cc
commit 7b71a72
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/chapter_multilayer-perceptrons/kaggle-house-price.md b/chapter_multilayer-perceptrons/kaggle-house-price.md
@@ -284,8 +284,8 @@ all_features.shape
 ```{.python .input}
 #@tab all
 n_train = train_data.shape[0]
-train_features = d2l.tensor(all_features[:n_train].values, dtype=d2l.float32)
-test_features = d2l.tensor(all_features[n_train:].values, dtype=d2l.float32)
+train_features = d2l.tensor(all_features[:n_train].values.astype(np.float32), dtype=d2l.float32)
+test_features = d2l.tensor(all_features[n_train:].values.astype(np.float32), dtype=d2l.float32)
 train_labels = d2l.tensor(
     train_data.SalePrice.values.reshape(-1, 1), dtype=d2l.float32)
 ```

diff --git a/chapter_multilayer-perceptrons/kaggle-house-price_origin.md b/chapter_multilayer-perceptrons/kaggle-house-price_origin.md
@@ -338,8 +338,8 @@ representation for training.
 ```{.python .input}
 #@tab all
 n_train = train_data.shape[0]
-train_features = d2l.tensor(all_features[:n_train].values, dtype=d2l.float32)
-test_features = d2l.tensor(all_features[n_train:].values, dtype=d2l.float32)
+train_features = d2l.tensor(all_features[:n_train].values.astype(np.float32), dtype=d2l.float32)
+test_features = d2l.tensor(all_features[n_train:].values.astype(np.float32), dtype=d2l.float32)
 train_labels = d2l.tensor(
     train_data.SalePrice.values.reshape(-1, 1), dtype=d2l.float32)
 ```