You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: official/wide_deep/README.md
+4-4Lines changed: 4 additions & 4 deletions
Original file line number
Diff line number
Diff line change
@@ -21,7 +21,7 @@ First make sure you've [added the models folder to your Python path](/official/#
21
21
The [Census Income Data Set](https://archive.ics.uci.edu/ml/datasets/Census+Income) that this sample uses for training is hosted by the [UC Irvine Machine Learning Repository](https://archive.ics.uci.edu/ml/datasets/). We have provided a script that downloads and cleans the necessary files.
22
22
23
23
```
24
-
python data_download.py
24
+
python census_dataset.py
25
25
```
26
26
27
27
This will download the files to `/tmp/census_data`. To change the directory, set the `--data_dir` flag.
@@ -30,12 +30,12 @@ This will download the files to `/tmp/census_data`. To change the directory, set
30
30
You can run the code locally as follows:
31
31
32
32
```
33
-
python wide_deep.py
33
+
python census_main.py
34
34
```
35
35
36
36
The model is saved to `/tmp/census_model` by default, which can be changed using the `--model_dir` flag.
37
37
38
-
To run the *wide* or *deep*-only models, set the `--model_type` flag to `wide` or `deep`. Other flags are configurable as well; see `wide_deep.py` for details.
38
+
To run the *wide* or *deep*-only models, set the `--model_type` flag to `wide` or `deep`. Other flags are configurable as well; see `census_main.py` for details.
39
39
40
40
The final accuracy should be over 83% with any of the three model types.
After the model finishes training, use [`saved_model_cli`](https://www.tensorflow.org/guide/saved_model#cli_to_inspect_and_execute_savedmodel) to inspect and execute the SavedModel.
0 commit comments