From a06670dac96977aff47f97c1496de54a939d5160 Mon Sep 17 00:00:00 2001 From: Abhiram V <61599526+Anon-Artist@users.noreply.github.com> Date: Tue, 19 Jan 2021 00:16:20 +0530 Subject: [PATCH] Update plots.py with yaml.SafeLoader (#1968) --- utils/plots.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/plots.py b/utils/plots.py index 47cd70776005..4765069e0377 100644 --- a/utils/plots.py +++ b/utils/plots.py @@ -301,7 +301,7 @@ def plot_labels(labels, save_dir=Path(''), loggers=None): def plot_evolution(yaml_file='data/hyp.finetune.yaml'): # from utils.plots import *; plot_evolution() # Plot hyperparameter evolution results in evolve.txt with open(yaml_file) as f: - hyp = yaml.load(f, Loader=yaml.FullLoader) + hyp = yaml.load(f, Loader=yaml.SafeLoader) x = np.loadtxt('evolve.txt', ndmin=2) f = fitness(x) # weights = (f - f.min()) ** 2 # for weighted results