我正在和scala合作一个Spark项目。我想训练一个模型,它可以是k_means,gaussian_mixture,逻辑回归,naive_bayes等,但我不能定义一个通用模型作为返回类型。由于这些算法的类型不同,如GaussianMixtureModel,KMeansModel等,我找不到任何逻辑方法来返回这个经过训练的模型。
以下是该项目的代码摘要:
model.model_algorithm match {
case "k_means" =>
val model_k_means = k_means(data, parameters)