Unverified 提交 028b7cdb authored 作者: Michael Ben ami's avatar Michael Ben ami 提交者: GitHub

fix_reading_nan_in_evolve (#10358)

when there is `nan` in evolve.csv pandas read it as str remove the space before fix that Signed-off-by: 's avatarMichael Ben ami <31584614+mbenami@users.noreply.github.com> Signed-off-by: 's avatarMichael Ben ami <31584614+mbenami@users.noreply.github.com> Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 b412696f
...@@ -1036,7 +1036,7 @@ def print_mutation(keys, results, hyp, save_dir, bucket, prefix=colorstr('evolve ...@@ -1036,7 +1036,7 @@ def print_mutation(keys, results, hyp, save_dir, bucket, prefix=colorstr('evolve
# Save yaml # Save yaml
with open(evolve_yaml, 'w') as f: with open(evolve_yaml, 'w') as f:
data = pd.read_csv(evolve_csv) data = pd.read_csv(evolve_csv, skipinitialspace=True)
data = data.rename(columns=lambda x: x.strip()) # strip keys data = data.rename(columns=lambda x: x.strip()) # strip keys
i = np.argmax(fitness(data.values[:, :4])) # i = np.argmax(fitness(data.values[:, :4])) #
generations = len(data) generations = len(data)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论