add classification with training data
Merge pull request #4 from cboss24/machineLearing

add classification with training data