From 5a9c5c1d3b02c8796e6af153d57af5821d8db0d6 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 13 Sep 2020 14:08:06 -0700 Subject: [PATCH] add mosaic and warmup to hyperparameters (#931) --- utils/general.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/general.py b/utils/general.py index eec8202e6bb7..b3253b112dec 100755 --- a/utils/general.py +++ b/utils/general.py @@ -1210,13 +1210,13 @@ def plot_evolution(yaml_file='data/hyp.finetune.yaml'): # from utils.general im x = np.loadtxt('evolve.txt', ndmin=2) f = fitness(x) # weights = (f - f.min()) ** 2 # for weighted results - plt.figure(figsize=(10, 10), tight_layout=True) + plt.figure(figsize=(10, 12), tight_layout=True) matplotlib.rc('font', **{'size': 8}) for i, (k, v) in enumerate(hyp.items()): y = x[:, i + 7] # mu = (y * weights).sum() / weights.sum() # best weighted result mu = y[f.argmax()] # best single result - plt.subplot(5, 5, i + 1) + plt.subplot(6, 5, i + 1) plt.scatter(y, f, c=hist2d(y, f, 20), cmap='viridis', alpha=.8, edgecolors='none') plt.plot(mu, f.max(), 'k+', markersize=15) plt.title('%s = %.3g' % (k, mu), fontdict={'size': 9}) # limit to 40 characters