abstract class Estimator[In, TrainIn, Out, TrainOut, Loss, EvalIn] extends AnyRef
Linear Supertypes
Known Subclasses
Ordering
- Alphabetic
- By Inheritance
Inherited
- Estimator
- AnyRef
- Any
- Hide All
- Show All
Visibility
- Public
- Protected
Abstract Value Members
- abstract def evaluate[TrainInD, TrainInS](data: () => Dataset[TrainIn], metrics: Seq[Metric[EvalIn, ops.Output[Float]]], maxSteps: Long = -1L, saveSummaries: Boolean = true, name: String = null)(implicit evOutputToDataType: Aux[TrainIn, TrainInD], evOutputToShape: Aux[TrainIn, TrainInS]): Seq[tensors.Tensor[Float]]
- Annotations
- @throws(scala.this.throws.<init>$default$1[org.platanios.tensorflow.api.core.exception.InvalidArgumentException])
- abstract def infer[InV, InD, InS, OutV, OutD, OutS, InferIn, InferOut](input: () => InferIn)(implicit evOutputToDataTypeIn: Aux[In, InD], evOutputToDataTypeOut: Aux[Out, OutD], evOutputToShapeIn: Aux[In, InS], evOutputToShapeOut: Aux[Out, OutS], evOutputToTensorIn: Aux[In, InV], evOutputToTensorOut: Aux[Out, OutV], ev: SupportedInferInput[In, InV, OutV, InferIn, InferOut], evOutputToTensorInOut: Aux[(In, Out), (InV, OutV)]): InferOut
- abstract def train[TrainInD, TrainInS](data: () => Dataset[TrainIn], stopCriteria: StopCriteria = StopCriteria())(implicit evOutputToDataType: Aux[TrainIn, TrainInD], evOutputToShape: Aux[TrainIn, TrainInS]): Unit
Concrete Value Members
- final def !=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
- final def ##: Int
- Definition Classes
- AnyRef → Any
- final def ==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
- final def asInstanceOf[T0]: T0
- Definition Classes
- Any
- def checkpointConfig: CheckpointConfig
- def clone(): AnyRef
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.CloneNotSupportedException]) @native() @HotSpotIntrinsicCandidate()
- val configuration: Configuration
- val configurationBase: Configuration
- Attributes
- protected
- val deviceFunction: Option[(OpSpecification) => String]
- final def eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
- def equals(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef → Any
- final def getClass(): Class[_ <: AnyRef]
- Definition Classes
- AnyRef → Any
- Annotations
- @native() @HotSpotIntrinsicCandidate()
- def getOrCreateSaver(): Option[Saver]
- Attributes
- protected
- def hashCode(): Int
- Definition Classes
- AnyRef → Any
- Annotations
- @native() @HotSpotIntrinsicCandidate()
- final def isInstanceOf[T0]: Boolean
- Definition Classes
- Any
- val modelFunction: ModelFunction[In, TrainIn, Out, TrainOut, Loss, EvalIn]
- Attributes
- protected
- final def ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
- final def notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native() @HotSpotIntrinsicCandidate()
- final def notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native() @HotSpotIntrinsicCandidate()
- def randomSeed: Option[Int]
- def saveEvaluationSummaries(step: Long, metrics: Seq[Metric[EvalIn, ops.Output[Float]]], metricValues: Seq[tensors.Tensor[Float]], name: String = null): Unit
- Attributes
- protected
- def sessionConfig: Option[SessionConfig]
- final def synchronized[T0](arg0: => T0): T0
- Definition Classes
- AnyRef
- def toString(): String
- Definition Classes
- AnyRef → Any
- final def wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.InterruptedException])
- final def wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.InterruptedException]) @native()
- final def wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws(classOf[java.lang.InterruptedException])
- def workingDir: Option[Path]