added more information on the wide path and the deep path
Browse files
README.md
CHANGED
@@ -14,9 +14,15 @@ metrics:
|
|
14 |
|
15 |
A wide & deep neural network trained on the California Housing dataset.
|
16 |
|
17 |
-
It takes eight
|
18 |
|
19 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
20 |
|
21 |
![](https://raw.githubusercontent.com/sambitmukherjee/handson-ml3-pytorch/main/chapter10/Figure_10-14.png)
|
22 |
|
|
|
14 |
|
15 |
A wide & deep neural network trained on the California Housing dataset.
|
16 |
|
17 |
+
It takes eight features: `'MedInc'`, `'HouseAge'`, `'AveRooms'`, `'AveBedrms'`, `'Population'`, `'AveOccup'`, `'Latitude'` and `'Longitude'`. It predicts `'MedHouseVal'`.
|
18 |
|
19 |
+
The first five features (`'MedInc'`, `'HouseAge'`, `'AveRooms'`, `'AveBedrms'` and `'Population'`) flow through the wide path.
|
20 |
+
|
21 |
+
The last six features (`'AveRooms'`, `'AveBedrms'`, `'Population'`, `'AveOccup'`, `'Latitude'` and `'Longitude'`) flow through the deep path.
|
22 |
+
|
23 |
+
Note: The features `'AveRooms'`, `'AveBedrms'` and `'Population'` flow through both the wide path and the deep path.
|
24 |
+
|
25 |
+
This model is a PyTorch adaptation of the TensorFlow model in Chapter 10 of Aurelien Geron's book 'Hands-On Machine Learning with Scikit-Learn, Keras, and TensorFlow'.
|
26 |
|
27 |
![](https://raw.githubusercontent.com/sambitmukherjee/handson-ml3-pytorch/main/chapter10/Figure_10-14.png)
|
28 |
|