From 8a9595c6150b564b61433a63cf75ef441bbc33d9 Mon Sep 17 00:00:00 2001 From: junos Date: Thu, 18 May 2023 22:53:58 +0200 Subject: [PATCH] Better handling of input filename again. --- exploration/ml_pipeline_classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/exploration/ml_pipeline_classification.py b/exploration/ml_pipeline_classification.py index d005785..84301b5 100644 --- a/exploration/ml_pipeline_classification.py +++ b/exploration/ml_pipeline_classification.py @@ -55,7 +55,7 @@ print("SEGMENT_LENGTH: " + SEGMENT_LENGTH) TARGET_VARIABLE = "JCQ_job_control" print("TARGET_VARIABLE: " + TARGET_VARIABLE) -if "appraisal" in TARGET_VARIABLE: +if ("appraisal" in TARGET_VARIABLE) and ("stressfulness" in TARGET_VARIABLE): TARGET_VARIABLE += "_" TARGET_VARIABLE += SEGMENT_TYPE