diff --git a/datasets/raise/attribute_source_camera.py b/datasets/raise/attribute_source_camera.py
index 13697da..de236cc 100755
--- a/datasets/raise/attribute_source_camera.py
+++ b/datasets/raise/attribute_source_camera.py
@@ -18,6 +18,7 @@ IMAGES_CAMERAS_FOLDER = {
     'Rafael 23/04/24': 'rafael/230424',
 }
 TRAINING_PORTION = 0.5
+PREDICT_ONLY_ON_WHOLE_TRAINING_SET = False
 
 setting = ','.join([escapeFilePath(imageCameraFolder) for imageCameraFolder in IMAGES_CAMERAS_FOLDER]) + f'_{DENOISER}'
 
@@ -109,7 +110,7 @@ for computeExtremes in tqdm(([True] if minColor is None or maxColor is None else
 
             cameraIterativeMean = camerasIterativeMean[camera]
             cameraIterativeMean.add(imagePrnuEstimateNpArray)
-            if cameraIndex == numberOfCameras - 1:
+            if cameraIndex == numberOfCameras - 1 and (not PREDICT_ONLY_ON_WHOLE_TRAINING_SET or cameraTrainingImageIndex == numberOfTrainingImages - 1):
                 numberOfTrainingImagesAccuracy = 0
                 print(f'{numberOfTestingImages=} {numberOfCameras=}')
                 # Loop over each camera testing image folder.