Skip to content

Commit

Permalink
add mosaic and warmup to hyperparameters (#931)
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher committed Sep 13, 2020
1 parent f1c63e2 commit 5a9c5c1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions utils/general.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 5a9c5c1

Please sign in to comment.