[Examples/TensorFlow] minor refactoring to allow compatible datasets to work
This PR removes the hard-coded "wikitext" values from the scripts so that they can be used in conjunction with any compatible dataset. 

@Rocketknight1 FYI.
_The documentation is not available anymore as the PR was closed or merged._