21 lines
389 B
Python
21 lines
389 B
Python
import pandas as pd
|
|
import matplotlib.pyplot as plt
|
|
|
|
df = pd.read_csv("benchmark_results.csv")
|
|
names = df.name
|
|
df = df.drop("name", axis=1)
|
|
|
|
plt.figure()
|
|
plt.bar(names, df.time)
|
|
plt.title("Training times")
|
|
plt.xlabel("Classifier")
|
|
plt.ylabel("Time (s)")
|
|
|
|
plt.figure()
|
|
plt.bar(names, df.accuracy)
|
|
plt.title("Prediction Accuracy")
|
|
plt.xlabel("Classifier")
|
|
plt.ylabel("Accuracy")
|
|
|
|
plt.show()
|