actsnclass.DataBase.classify

DataBase.classify(method='RandomForest', screen=False, n_est=1000, seed=42, max_depth=None, n_jobs=1, return_model=False)

Apply a machine learning classifier.

Populate properties: predicted_class and class_prob

Parameters:

• method (str (optional)) – Chosen classifier. The current implementation on accepts RandomForest and XGBoost.

• n_est (int (optional)) – Number of trees. Default is 1000.

• screen (bool (optional)) – If True, print debug comments on screen. Default is False.

• seed (int (optional)) – Random seed. Default is 42.

• max_depth (None or int (optional)) – The maximum depth of the tree. Default is None.

• n_jobs (int (optional)) – Number of cores used to train the model. Default is 1.

• return_model (bool (optional)) – If True, returned the trained model as a third output. Default is False.