From 223cbde49f1feaec39bb158d937c9bba2ad150fd Mon Sep 17 00:00:00 2001
From: tzirakis <ptzirakis88@yahoo.com>
Date: Thu, 18 Apr 2019 14:35:54 +0100
Subject: [PATCH] Update data_generator.py

---
 data_generator.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/data_generator.py b/data_generator.py
index 87162ab..6cf3cf6 100644
--- a/data_generator.py
+++ b/data_generator.py
@@ -12,10 +12,11 @@
 
 root_dir = Path('path_of_RECOLA')
 
+# Inser the numbers for each video for the RECOLA Dataset
 portion_to_id = dict(
-    train = [25, 15, 16 ,17 ,18 ,21 ,23 ,37 ,39 ,41 ,46 ,50 ,51 ,55 ,56, 60], # 25
-    valid = [14, 19, 24, 26, 28, 30, 34 ,40, 42, 43, 44, 45, 52, 64, 65],
-    test  = [54, 53, 13, 20, 22, 32, 38, 47, 48, 49, 57, 58, 59, 62, 63] # 54, 53
+    train = [], 
+    valid = [],
+    test  = [] 
 )
 
 def get_samples(subject_id):