tried to make algorithm more detailed and got failed results
edited training method and reverted predict method; results up to 96%