Merge branch 'feature/cara_pipeline_model' into feature/cara_pipeline

# Conflicts: # src/main/scala/io/github/jsarni/CaraModel.scala # src/test/scala/io/github/jsarni/CaraModelTest.scala
jsarni · Jun 18, 2021 · 36f534b · 36f534b
2 parents a532e5c + 8f830ae
commit 36f534b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/scala/io/github/jsarni/CaraModel.scala b/src/main/scala/io/github/jsarni/CaraModel.scala
@@ -27,7 +27,7 @@ final class CaraModel(yamlPath: String, dataset: Dataset[_], savePath: String)(i
     val model = PipelineModel.load(savePath)
     model.transform(dataset)
   }
-  
+
   private def generateModel(caraPipeline: CaraPipeline) : Try[Pipeline] = Try {
     val pipeline = caraPipeline.pipeline
     val evaluator = caraPipeline.evaluator
@@ -44,7 +44,7 @@ final class CaraModel(yamlPath: String, dataset: Dataset[_], savePath: String)(i
 
         crossValidatorModel.getClass.getMethod(methodeName, paramValue.getClass )
           .invoke(crossValidatorModel,paramValue.asInstanceOf[java.lang.Integer])
-        
+
         new Pipeline().setStages(Array(crossValidatorModel))
       }
       case "TrainValidationSplit" => {