Improve docs for initialEpochs

[pvaneck]

This PR fixes fixes the epochs docstring for fitDataset to match what was changed here: 4393140

Some other minor fixes were added as well.

---

This change is 

[tafsiri]

Thanks for doing this (the little fixes do help!). I just have one request about the where to locate the information about the relationship between initialEpoch and epochs.

Reviewed 10 of 10 files at r1.
Reviewable status: 0 of 1 approvals obtained (waiting on @pvaneck)

---

tfjs-layers/src/engine/training_dataset.ts, line 42 at r1 (raw file):

  /**
   * The number of times to iterate over the training dataset.
   * Note that when used with `initialEpoch`, epochs is the index of the

I think this elaboration might better belong to the docs for initialEpochs to help guide the use of that param. This makes it so that you only think about this when you start to use intialEpochs.

Could you change that here and in the file referenced in your other commit.

[pvaneck]

Reviewable status: 0 of 1 approvals obtained (waiting on @tafsiri)

---

tfjs-layers/src/engine/training_dataset.ts, line 42 at r1 (raw file):

Previously, tafsiri (Yannick Assogba) wrote…

I think this elaboration might better belong to the docs for initialEpochs to help guide the use of that param. This makes it so that you only think about this when you start to use intialEpochs.

Could you change that here and in the file referenced in your other commit.

Good idea. Updated!

[tafsiri]

Reviewed 2 of 2 files at r2.
Reviewable status: complete! 1 of 1 approvals obtained

[tafsiri]

Thanks!