From b89cae34b45d1a9eb291aae7cc66912cbb6c5f35 Mon Sep 17 00:00:00 2001 From: Jessy Lauer <30733203+jeylau@users.noreply.github.com> Date: Mon, 12 Jun 2023 15:05:02 +0200 Subject: [PATCH] Pass train/test indices to create_multianimaltraining_dataset --- .../generate_training_dataset/trainingsetmanipulation.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/deeplabcut/generate_training_dataset/trainingsetmanipulation.py b/deeplabcut/generate_training_dataset/trainingsetmanipulation.py index a51826931e..623b336a80 100755 --- a/deeplabcut/generate_training_dataset/trainingsetmanipulation.py +++ b/deeplabcut/generate_training_dataset/trainingsetmanipulation.py @@ -845,7 +845,10 @@ def create_training_dataset( ) create_multianimaltraining_dataset( - config, num_shuffles, Shuffles, net_type=net_type + config, num_shuffles, Shuffles, + net_type=net_type, + trainIndices=trainIndices, + testIndices=testIndices, ) else: scorer = cfg["scorer"]