Benchmarking script for cuML RF->FIL conversion

rf-fil.py

from time import perf_counter

from cuml import using_output_type
from cuml.datasets import make_classification
from cuml.ensemble import RandomForestClassifier
from cuml.metrics import accuracy_score

with using_output_type('cupy'):
    data, labels = make_classification(
        n_samples=int(1e5),
        n_features=20,
        n_informative=18,
        n_classes=2,
        random_state=0
    )

n_trees = 300

classifier = RandomForestClassifier(n_estimators=n_trees)
classifier.fit(data, labels)

# start = perf_counter()
preds = classifier.predict(data)
# runtime = perf_counter() - start
print(accuracy_score(preds, labels))
# print(f"Run 1: {runtime}")
# 
# start = perf_counter()
# classifier.predict(data)
# runtime = perf_counter() - start
# print(f"Run 2: {runtime}")