changed sklearn version
fixed bugs
added docs
added python notebook
dockerfile
balanced
initial commit
