fixed errors - converted train_features and test_features to tensor, …

[gitshiva]

…then updated Horsepower references to specific columns in array, otherwise this nb was failing due to data type mismatches

closes: https://github.com/tensorflow/docs/pull/2313

[github-actions[bot]]

Preview

Preview and run these notebook edits with Google Colab:

• site/en/tutorials/keras/regression.ipynb

Rendered notebook diffs available on ReviewNB.com.

Format and style

Use the TensorFlow docs notebook tools to format for consistent source diffs and lint for style:

$ python3 -m pip install -U --user git+https://github.com/tensorflow/docs

$ python3 -m tensorflow_docs.tools.nbfmt notebook.ipynb

$ python3 -m tensorflow_docs.tools.nblint --arg=repo:tensorflow/docs notebook.ipynb

If commits are added to the pull request, synchronize your local branch: git pull origin master

[MarkDaoust]

Hi, thanks for pointing this out and taking the time to send a PR.

I always prefer referencing columns by name instead of index, how about we do this:

https://github.com/gitshiva/tensorflow-docs/pull/2