RF_Classification

source

RF_Classification(
   data_obj: Classification_Data
)

Methods:

.run_classifier

source

.run_classifier(
   data_obj
)

---

Initialize the model, train (if not loaded) and evaluate on test data

Args

• data_obj : Classification_Data object

Returns

data_obj with modified variables

.train_model

source

.train_model()

---

Initializes and trains the random forest

Returns

RandomForestClassifier

.plot

source

.plot(
   data_obj
)

---

Creates the plots

Args

• data_obj : Classification_Data object

Returns

data_object with modified variables