Abhiram V commited on
Commit
35400dc
1 Parent(s): b5d851d

Update plots.py with yaml.SafeLoader (#1968)

Browse files
Files changed (1) hide show
  1. 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.FullLoader)
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