actsnclass.random_forest

actsnclass.random_forest(train_features: numpy.array, train_labels: numpy.array, test_features: numpy.array, **kwargs)

Random Forest classifier.

Parameters:
  • train_features (np.array) – Training sample features.
  • train_labels (np.array) – Training sample classes.
  • test_features (np.array) – Test sample features.
  • kwargs (extra parameters) – All keywords required by sklearn.ensemble.RandomForestClassifier function.
Returns:

  • predictions (np.array) – Predicted classes.
  • prob (np.array) – Classification probability for all objects, [pIa, pnon-Ia].