From 18d125cfa6729f291adec4bb7437acba38c3ba4c Mon Sep 17 00:00:00 2001
From: Devin Ha <t.ha@tu-berlin.de>
Date: Sat, 3 Jun 2023 17:23:23 +0200
Subject: [PATCH] Resolve saving bug with multilabel parameter

---
 .../johnsnowlabs/nlp/HasClassifierActivationProperties.scala  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/scala/com/johnsnowlabs/nlp/HasClassifierActivationProperties.scala b/src/main/scala/com/johnsnowlabs/nlp/HasClassifierActivationProperties.scala
index 1f1dc4bdf48ce3..9d92e7a02482e6 100644
--- a/src/main/scala/com/johnsnowlabs/nlp/HasClassifierActivationProperties.scala
+++ b/src/main/scala/com/johnsnowlabs/nlp/HasClassifierActivationProperties.scala
@@ -16,7 +16,7 @@
 
 package com.johnsnowlabs.nlp
 
-import org.apache.spark.ml.param.{FloatParam, Param}
+import org.apache.spark.ml.param.{BooleanParam, FloatParam, Param}
 
 trait HasClassifierActivationProperties extends ParamsAndFeaturesWritable {
 
@@ -47,7 +47,7 @@ trait HasClassifierActivationProperties extends ParamsAndFeaturesWritable {
     *
     * @group param
     */
-  val multilabel: Param[Boolean] = new Param(
+  val multilabel: BooleanParam = new BooleanParam(
     this,
     "multilabel",
     "Whether or not the result should be multi-class (the sum of all probabilities is 1.0) or multi-label (each label has a probability between 0.0 to 1.0). Default is False i.e. multi-class")