Class NeuralNetwork

constructor

• new NeuralNetwork(
      graph: ComputationalGraph,
      optimizer?:
          | "sgd"
          | "adam"
          | "momentum"
          | "adagrad"
          | "rmsprop"
          | "adadelta"
          | "rmspropgraves"
          | "smorms3"
          | "adamax"
          | "nadam"
          | "santae"
          | "santasss"
          | "amsgrad"
          | "adabound"
          | "amsbound"
          | "adabelief",
  ): NeuralNetwork

• Parameters

  • graph: ComputationalGraph

    Graph of a network
  • Optionaloptimizer:
        | "sgd"
        | "adam"
        | "momentum"
        | "adagrad"
        | "rmsprop"
        | "adadelta"
        | "rmspropgraves"
        | "smorms3"
        | "adamax"
        | "nadam"
        | "santae"
        | "santasss"
        | "amsgrad"
        | "adabound"
        | "amsbound"
        | "adabelief"

    Optimizer of the network

  Returns NeuralNetwork

_graph

_opt

_opt_managers

_optimizer

_getDatasize

• _getDatasize(x: any): any

• Parameters

  • x: any

  Returns any

calc

• calc(
      x:
          | Tensor
          | Matrix
          | (number | number[])[][]
          | { [key: string]: Tensor | Matrix | (number | number[])[][] },
      t?: Matrix,
      out?: string[],
      options?: object,
  ): Matrix | { [key: string]: Matrix }

• Returns calculated values.

  Parameters

  • x:
        | Tensor
        | Matrix
        | (number | number[])[][]
        | { [key: string]: Tensor | Matrix | (number | number[])[][] }

    Input value
  • Optionalt: Matrix

    Supervised value
  • Optionalout: string[]

    Name of node from which to get output
  • Optionaloptions: object

    Option

  Returns Matrix | { [key: string]: Matrix }

  Calculated values

copy

• copy(): NeuralNetwork

• Returns a copy of this.

  Returns NeuralNetwork

  Copied network

fit

• fit(
      x:
          | Tensor
          | Matrix
          | (number | number[])[][]
          | { [key: string]: Tensor | Matrix | (number | number[])[][] },
      t: Matrix | number[][],
      epoch?: number,
      learning_rate?: number,
      batch_size?: number,
      options?: object,
  ): number[]

• Fit model.

  Parameters

  • x:
        | Tensor
        | Matrix
        | (number | number[])[][]
        | { [key: string]: Tensor | Matrix | (number | number[])[][] }

    Training data
  • t: Matrix | number[][]

    Target values
  • Optionalepoch: number

    Iteration count
  • Optionallearning_rate: number

    Learning rate
  • Optionalbatch_size: number

    Batch size
  • Optionaloptions: object

    Option

  Returns number[]

  Loss value

grad

• grad(e?: Matrix): Matrix

• Returns gradient values.

  Parameters

  • Optionale: Matrix

    Input of gradient

  Returns Matrix

  Output of gradient

predict

• predict(
      x:
          | Tensor
          | Matrix
          | (number | number[])[][]
          | { [key: string]: Tensor | Matrix | (number | number[])[][] },
  ): number[][]

• Returns predicted values.

  Parameters

  • x:
        | Tensor
        | Matrix
        | (number | number[])[][]
        | { [key: string]: Tensor | Matrix | (number | number[])[][] }

    Sample data

  Returns number[][]

  Predicted values

toObject

• toObject(): LayerObject[]

• Returns object representation.

  Returns LayerObject[]

  Object represented this neuralnetwork

update

• update(learning_rate: number): void

• Update model parameters.

  Parameters

  • learning_rate: number

    Learning rate

  Returns void

StaticfromObject

• fromObject(
      layers: LayerObject[],
      loss?: string,
      optimizer?:
          | "sgd"
          | "adam"
          | "momentum"
          | "adagrad"
          | "rmsprop"
          | "adadelta"
          | "rmspropgraves"
          | "smorms3"
          | "adamax"
          | "nadam",
  ): NeuralNetwork

• Returns neuralnetwork.

  Parameters

  • layers: LayerObject[]

    Network layers
  • Optionalloss: string

    Loss name
  • Optionaloptimizer:
        | "sgd"
        | "adam"
        | "momentum"
        | "adagrad"
        | "rmsprop"
        | "adadelta"
        | "rmspropgraves"
        | "smorms3"
        | "adamax"
        | "nadam"

    Optimizer of the network

  Returns NeuralNetwork

  Created Neuralnetwork

StaticfromONNX

• fromONNX(buffer: Uint8Array | ArrayBuffer | File): Promise<NeuralNetwork>

• Load onnx model.

  Parameters

  • buffer: Uint8Array | ArrayBuffer | File

    File

  Returns Promise<NeuralNetwork>

  Loaded NeuralNetwork