experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixing typo in documentation.

Browse Source 
epochs is not the third argument to fit() (it's batch_size)

PiperOrigin-RevId: 316284462
Change-Id: Icf223f0617d814edd1f3801cafa08fb609478874
This commit is contained in:
A. Unique TensorFlower 2020-06-13 14:19:04 -07:00 committed by TensorFlower Gardener
parent 8e6cb257fe
commit 52d1e72c66
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/keras/premade/linear.py
View File

@ -41,7 +41,7 @@ class LinearModel(training.Model):
```python ```python
model = LinearModel() model = LinearModel()
model.compile(optimizer='sgd', loss='mse') model.compile(optimizer='sgd', loss='mse')
model.fit(x, y, epochs) model.fit(x, y, epochs=epochs)
``` ```
This model accepts sparse float inputs as well: This model accepts sparse float inputs as well: