Abhiram V
commited on
Commit
•
35400dc
1
Parent(s):
b5d851d
Update plots.py with yaml.SafeLoader (#1968)
Browse files- utils/plots.py +1 -1
utils/plots.py
CHANGED
@@ -301,7 +301,7 @@ def plot_labels(labels, save_dir=Path(''), loggers=None):
|
|
301 |
def plot_evolution(yaml_file='data/hyp.finetune.yaml'): # from utils.plots import *; plot_evolution()
|
302 |
# Plot hyperparameter evolution results in evolve.txt
|
303 |
with open(yaml_file) as f:
|
304 |
-
hyp = yaml.load(f, Loader=yaml.
|
305 |
x = np.loadtxt('evolve.txt', ndmin=2)
|
306 |
f = fitness(x)
|
307 |
# weights = (f - f.min()) ** 2 # for weighted results
|
|
|
301 |
def plot_evolution(yaml_file='data/hyp.finetune.yaml'): # from utils.plots import *; plot_evolution()
|
302 |
# Plot hyperparameter evolution results in evolve.txt
|
303 |
with open(yaml_file) as f:
|
304 |
+
hyp = yaml.load(f, Loader=yaml.SafeLoader)
|
305 |
x = np.loadtxt('evolve.txt', ndmin=2)
|
306 |
f = fitness(x)
|
307 |
# weights = (f - f.min()) ** 2 # for weighted results
|