Update README.md
Modify "load" function.
Add keyword.
Add "train" sample.