org.apache.spark.ml.NamespaceInjections.scala Maven / Gradle / Ivy
The newest version!
// Copyright (C) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See LICENSE in project root for information.
package org.apache.spark.ml
import org.apache.spark.ml.image.ImageSchema
import org.apache.spark.ml.util.Identifiable
import org.apache.spark.sql.Row
object NamespaceInjections {
def pipelineModel[T <: Transformer](stages: Array[T]): PipelineModel = {
new PipelineModel(Identifiable.randomUID("PipelineModel"), stages.asInstanceOf[Array[Transformer]])
}
}
object ImageInjections {
def decode(origin: String, bytes: Array[Byte]): Option[Row] = ImageSchema.decode(origin, bytes)
}