Вот моя модель:
X, Y_segmentation, Y_classification, X_filenames, Y_filenames = shuffle(
np.array(X), np.array(Y_segmentation), np.array(Y_classification),
X_filenames, Y_filenames, random_state=42
)
# Train-test split
X_train, X_val, Y_seg_train,...