Fixed parsing data
Fixed getting the right features out
Back to 10
Rid of train test
Changed number of iterations
Put back one-hotting
Merge pull request #21 from sihrc/Fix

Fix