Update training_guide.md
Browse files- training_guide.md +1 -0
training_guide.md
CHANGED
|
@@ -79,6 +79,7 @@ def transform(example_batch):
|
|
| 79 |
|
| 80 |
|
| 81 |
return inputs
|
|
|
|
| 82 |
```
|
| 83 |
Well, this code might not look pretty, but it gets the job done! As for the images (inputs), we resize them to 224x224 and flatten them out. Now, for the labels (target), we're transforming them into a multi-hot format. Why, you ask? Because I like it that way, and it's simple.
|
| 84 |
|
|
|
|
| 79 |
|
| 80 |
|
| 81 |
return inputs
|
| 82 |
+
prepared_dataset = dataset.with_transform(transform)
|
| 83 |
```
|
| 84 |
Well, this code might not look pretty, but it gets the job done! As for the images (inputs), we resize them to 224x224 and flatten them out. Now, for the labels (target), we're transforming them into a multi-hot format. Why, you ask? Because I like it that way, and it's simple.
|
| 85 |
|