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