Update kfold-cross-validation.md (#17332)

Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
fix-yolo-solutions-cli-ci
Mahdi Amrollahi 3 weeks ago committed by GitHub
parent da15e27a4d
commit d049e22769
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 1
      docs/en/guides/kfold-cross-validation.md

@ -263,6 +263,7 @@ fold_lbl_distrb.to_csv(save_path / "kfold_label_distribution.csv")
for k in range(ksplit): for k in range(ksplit):
dataset_yaml = ds_yamls[k] dataset_yaml = ds_yamls[k]
model = YOLO(weights_path, task="detect")
model.train(data=dataset_yaml, epochs=epochs, batch=batch, project=project) # include any train arguments model.train(data=dataset_yaml, epochs=epochs, batch=batch, project=project) # include any train arguments
results[k] = model.metrics # save output metrics for further analysis results[k] = model.metrics # save output metrics for further analysis
``` ```

Loading…
Cancel
Save