Options
All
  • Public
  • Public/Protected
  • All
Menu

Class NeuralNetwork<TNetParams>

Type parameters

  • TNetParams

Hierarchy

Index

Constructors

  • new NeuralNetwork<TNetParams>(name: string): NeuralNetwork<TNetParams>

Properties

_name: any

Accessors

  • get isLoaded(): boolean
  • get paramMappings(): ParamMapping[]
  • get params(): undefined | TNetParams

Methods

  • dispose(throwOnRedispose?: boolean): void
  • extractWeights(weights: Float32Array): void
  • freeze(): void
  • getFrozenParams(): { path: string; tensor: Tensor }[]
  • getParamFromPath(paramPath: string): Tensor
  • getParamList(): { path: string; tensor: Tensor }[]
  • getTrainableParams(): { path: string; tensor: Tensor }[]
  • load(weightsOrUrl: undefined | string | Float32Array): Promise<void>
  • Parameters

    • weightsOrUrl: undefined | string | Float32Array

    Returns Promise<void>

  • loadFromDisk(filePath: undefined | string): Promise<void>
  • loadFromUri(uri: undefined | string): Promise<void>
  • loadFromWeightMap(weightMap: NamedTensorMap): void
  • reassignParamFromPath(paramPath: string, tensor: Tensor): void
  • serializeParams(): Float32Array
  • variable(): void