com.intel.analytics.bigdl.utils.tf

Session

abstract class Session[T] extends AnyRef

Linear Supertypes
AnyRef, Any
Known Subclasses
BigDLSessionImpl
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. Session
  2. AnyRef
  3. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new Session()(implicit arg0: ClassTag[T])

Abstract Value Members

  1. abstract def predict(endPoints: Seq[String], isDataBatch: Boolean, batchSize: Int, sc: SparkContext): RDD[Activity]

    Predict data with tensorflow graph.

    Predict data with tensorflow graph. The data must hold in a queue

    endPoints
    isDataBatch

    if the model input is the batch

    batchSize

    batch size, which should be original batch size * total core number

    sc
    returns

  2. abstract def saveParameters(binFile: String): Session.this.type

    Dump varaible contents to a file

    Dump varaible contents to a file

    binFile
    returns

  3. abstract def train(endPoints: Seq[String], optMethod: OptimMethod[T], endWhen: Trigger, isDataBatch: Boolean, batchSize: Int, sc: SparkContext, loss: Option[String]): Session.this.type

    Train the tensorflow graph.

    Train the tensorflow graph. The model must be fed data with a queue

    endPoints
    optMethod
    endWhen
    isDataBatch

    if the model input is the batch

    batchSize

    batch size, which should be original batch size * total core number

    sc
    loss
    returns

  4. abstract def train(outputs: Seq[String], dataSet: DistributedDataSet[MiniBatch[T]], optMethod: OptimMethod[T], criterion: Criterion[T], endWhen: Trigger): Graph[T]

    Train the tensorflow graph

    Train the tensorflow graph

    outputs
    dataSet
    optMethod
    criterion
    endWhen
    returns

Concrete Value Members

  1. final def !=(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef

  2. final def !=(arg0: Any): Boolean

    Definition Classes
    Any

  3. final def ##(): Int

    Definition Classes
    AnyRef → Any

  4. final def ==(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef

  5. final def ==(arg0: Any): Boolean

    Definition Classes
    Any

  6. final def asInstanceOf[T0]: T0

    Definition Classes
    Any

  7. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  8. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef

  9. def equals(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any

  10. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )

  11. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any

  12. def hashCode(): Int

    Definition Classes
    AnyRef → Any

  13. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any

  14. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef

  15. final def notify(): Unit

    Definition Classes
    AnyRef

  16. final def notifyAll(): Unit

    Definition Classes
    AnyRef

  17. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef

  18. def toString(): String

    Definition Classes
    AnyRef → Any

  19. final def wait(): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  20. final def wait(arg0: Long, arg1: Int): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  21. final def wait(arg0: Long): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from AnyRef

Inherited from Any

Ungrouped