Fixed the issue of class attribute

trained on the entire dataset

Upload basic model

initial commit