optimus_random_forest.py

from optimus import Optimus
from optimus.ml.models import ML

op = Optimus()
ml = ML()

df_predict, rf_model = ml.random_forest(df_cancer, columns, "diagnosis")

['label',
     'diagnosis',
     'radius_mean',
     'texture_mean',
     'perimeter_mean',
     'area_mean',
     'smoothness_mean',
     'compactness_mean',
     'concavity_mean',
     'concave points_mean',
     'symmetry_mean',
     'fractal_dimension_mean',
     'features',
     'rawPrediction',
     'probability',
     'prediction']