All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.microsoft.ml.spark.cntk.CNTKFunctionParam.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 com.microsoft.ml.spark.cntk

import com.microsoft.CNTK.SerializableFunction
import com.microsoft.ml.spark.core.serialize.ComplexParam
import org.apache.spark.ml.param.Params

/** Param for ByteArray.  Needed as spark has explicit com.microsoft.ml.spark.core.serialize.params for many different
  * types but not ByteArray.
  */
class CNTKFunctionParam(parent: Params, name: String, doc: String,
                        isValid: SerializableFunction => Boolean)

  extends ComplexParam[SerializableFunction](parent, name, doc, isValid) {

  def this(parent: Params, name: String, doc: String) =
    this(parent, name, doc, {x => true})

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy