diff --git a/modules/machine_learning/classification.py b/modules/machine_learning/classification.py
index 18040b7ac575bfb1aa22619295d0e70c478e4a49..914771c7c4f5e88ce6ecac16237be056de9b98a4 100644
--- a/modules/machine_learning/classification.py
+++ b/modules/machine_learning/classification.py
@@ -16,8 +16,8 @@ from sklearn.tree import export_graphviz
 
 # The independent variables (features) we want to use:
 # feature_params = ['cld_temp_acha', 'supercooled_cloud_fraction', 'cld_reff_dcomp', 'cld_opd_dcomp', 'cld_cwp_dcomp']
-# feature_params = ['cld_temp_acha', 'supercooled_cloud_fraction', 'cld_reff_dcomp', 'cld_opd_dcomp']
-feature_params = ['cld_temp_acha', 'supercooled_cloud_fraction', 'cld_reff_acha', 'cld_opd_acha']
+feature_params = ['cld_temp_acha', 'supercooled_cloud_fraction', 'cld_reff_dcomp', 'cld_opd_dcomp']
+# feature_params = ['cld_temp_acha', 'supercooled_cloud_fraction', 'cld_reff_acha', 'cld_opd_acha']
 
 # The dependent variable (target)
 target_param = 'icing_intensity'