diff --git a/utils/plots.py b/utils/plots.py index c0d196188751..3ec70b62b49c 100644 --- a/utils/plots.py +++ b/utils/plots.py @@ -426,7 +426,7 @@ def plot_results(file='path/to/results.csv', dir=''): s = [x.strip() for x in data.columns] x = data.values[:, 0] for i, j in enumerate([1, 2, 3, 4, 5, 8, 9, 10, 6, 7]): - y = data.values[:, j] + y = data.values[:, j].astype('float') # y[y == 0] = np.nan # don't show zero values ax[i].plot(x, y, marker='.', label=f.stem, linewidth=2, markersize=8) ax[i].set_title(s[j], fontsize=12)