Implemented predict function, can print confusion matrix and parse learning_rate and threshold keywords