diff --git a/src/shogun/labels/BinaryLabelEncoder.h b/src/shogun/labels/BinaryLabelEncoder.h index 27e4b0ccce1..fd7be6c356e 100644 --- a/src/shogun/labels/BinaryLabelEncoder.h +++ b/src/shogun/labels/BinaryLabelEncoder.h @@ -33,7 +33,7 @@ namespace shogun { const auto result_vector = labs->as()->get_labels(); check_is_valid(result_vector); - if (print_warning && !can_convert_float_to_int(result_vector)) + if (!can_convert_float_to_int(result_vector)) { io::warn( "({}, {}) have been converted to (-1, 1).", diff --git a/src/shogun/labels/LabelEncoder.h b/src/shogun/labels/LabelEncoder.h index 8f5ef132df5..273ee992c53 100644 --- a/src/shogun/labels/LabelEncoder.h +++ b/src/shogun/labels/LabelEncoder.h @@ -59,9 +59,6 @@ namespace shogun return "LabelEncoder"; } - void set_print_warning(bool print_warning){ - print_warning = print_warning; - } protected: SGVector fit_impl(const SGVector& origin_vector) { @@ -116,7 +113,6 @@ namespace shogun }); } - bool print_warning = true; std::set unique_labels; std::unordered_map normalized_to_origin; }; diff --git a/src/shogun/labels/MulticlassLabelsEncoder.h b/src/shogun/labels/MulticlassLabelsEncoder.h index 02fc05b269e..ffc6382ec95 100644 --- a/src/shogun/labels/MulticlassLabelsEncoder.h +++ b/src/shogun/labels/MulticlassLabelsEncoder.h @@ -32,11 +32,11 @@ namespace shogun SGVector fit(const std::shared_ptr& labs) override { const auto result_vector = labs->as()->get_labels(); - if (print_warning && !can_convert_float_to_int(result_vector)) + if (!can_convert_float_to_int(result_vector)) { std::set s(result_vector.begin(), result_vector.end()); io::warn( - "{} have been converted to 0...{}", + "({}) have been converted to (0...{})", fmt::join(s, ", "), result_vector.vlen - 1); }