ROC Curve¶
In [105]:
from verticapy.learn.ensemble import RandomForestClassifier
# Binary Classification
model = RandomForestClassifier(name = "public.RF_titanic",
n_estimators = 20,
max_features = "auto",
max_leaf_nodes = 32,
sample = 0.7,
max_depth = 3,
min_samples_leaf = 5,
min_info_gain = 0.0,
nbins = 32)
model.fit("public.titanic", ["age", "fare", "gender"], "survived")
# Roc Curve
model.roc_curve(color = "#0073E7")
Out[105]:
