Made the following changes
1. Remove call to KNeighborsClassifier
2. Set default branching_factor to 50