Commit
·
5a9c5c1
1
Parent(s):
f1c63e2
add mosaic and warmup to hyperparameters (#931)
Browse files- utils/general.py +2 -2
utils/general.py
CHANGED
@@ -1210,13 +1210,13 @@ def plot_evolution(yaml_file='data/hyp.finetune.yaml'): # from utils.general im
|
|
1210 |
x = np.loadtxt('evolve.txt', ndmin=2)
|
1211 |
f = fitness(x)
|
1212 |
# weights = (f - f.min()) ** 2 # for weighted results
|
1213 |
-
plt.figure(figsize=(10,
|
1214 |
matplotlib.rc('font', **{'size': 8})
|
1215 |
for i, (k, v) in enumerate(hyp.items()):
|
1216 |
y = x[:, i + 7]
|
1217 |
# mu = (y * weights).sum() / weights.sum() # best weighted result
|
1218 |
mu = y[f.argmax()] # best single result
|
1219 |
-
plt.subplot(
|
1220 |
plt.scatter(y, f, c=hist2d(y, f, 20), cmap='viridis', alpha=.8, edgecolors='none')
|
1221 |
plt.plot(mu, f.max(), 'k+', markersize=15)
|
1222 |
plt.title('%s = %.3g' % (k, mu), fontdict={'size': 9}) # limit to 40 characters
|
|
|
1210 |
x = np.loadtxt('evolve.txt', ndmin=2)
|
1211 |
f = fitness(x)
|
1212 |
# weights = (f - f.min()) ** 2 # for weighted results
|
1213 |
+
plt.figure(figsize=(10, 12), tight_layout=True)
|
1214 |
matplotlib.rc('font', **{'size': 8})
|
1215 |
for i, (k, v) in enumerate(hyp.items()):
|
1216 |
y = x[:, i + 7]
|
1217 |
# mu = (y * weights).sum() / weights.sum() # best weighted result
|
1218 |
mu = y[f.argmax()] # best single result
|
1219 |
+
plt.subplot(6, 5, i + 1)
|
1220 |
plt.scatter(y, f, c=hist2d(y, f, 20), cmap='viridis', alpha=.8, edgecolors='none')
|
1221 |
plt.plot(mu, f.max(), 'k+', markersize=15)
|
1222 |
plt.title('%s = %.3g' % (k, mu), fontdict={'size': 9}) # limit to 40 characters
|