您可以创建自己的特征转换（使用udf或其他方法），然后重写spark的转换方法，并将其放入自己的操作中。
github上的spark代码为您提供了一些关于扩展transformer功能的可能性的见解，前提是您需要创建 Package 器对象。

override def transform(dataset: Dataset[_]): DataFrame = {
        transformSchema(dataset.schema, logging = true)
        val xModel = new feature.XModel()
         val xOp = udf {xModel.transform _ }
         dataset.withColumn($(outputCol), xOp(col($(inputCol))))
      }

其中xmodel和xop是抽象。模型将根据定义的操作相应地转换数据集。