Training and Predicting

HiClass adheres to the same API from the popular machine learning library scikit-learn. Hence, training is as easy as calling the fit method on the training data:

classifier.fit(X_train, Y_train)

Prediction is performed by calling the predict method on the test features:

predictions = classifier.predict(X_test)