Class NeuralNetwork<TNetParams>Abstract

constructor

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

• Type Parameters

  • TNetParams

  Parameters

  • name: string

  Returns NeuralNetwork<TNetParams>

_name

isLoaded

• get isLoaded(): boolean

• Returns boolean

paramMappings

• get paramMappings(): ParamMapping[]

• Returns ParamMapping[]

params

• get params(): undefined | TNetParams

• Returns undefined | TNetParams

dispose

• dispose(throwOnRedispose?): void

• Parameters

  • throwOnRedispose: boolean = true

  Returns void

extractWeights

• extractWeights(weights): void

• Parameters

  • weights: Float32Array

  Returns void

freeze

• freeze(): void

• Returns void

getFrozenParams

• getFrozenParams(): {
      path: string;
      tensor: Tensor<Rank>;
  }[]

• Returns {
      path: string;
      tensor: Tensor<Rank>;
  }[]

getParamFromPath

• getParamFromPath(paramPath): Tensor<Rank>

• Parameters

  • paramPath: string

  Returns Tensor<Rank>

getParamList

• getParamList(): {
      path: string;
      tensor: Tensor<Rank>;
  }[]

• Returns {
      path: string;
      tensor: Tensor<Rank>;
  }[]

getTrainableParams

• getTrainableParams(): {
      path: string;
      tensor: Tensor<Rank>;
  }[]

• Returns {
      path: string;
      tensor: Tensor<Rank>;
  }[]

load

• load(weightsOrUrl): Promise<void>

• Parameters

  • weightsOrUrl: undefined | string | Float32Array

  Returns Promise<void>

loadFromDisk

• loadFromDisk(filePath): Promise<void>

• Parameters

  • filePath: undefined | string

  Returns Promise<void>

loadFromUri

• loadFromUri(uri): Promise<void>

• Parameters

  • uri: undefined | string

  Returns Promise<void>

loadFromWeightMap

• loadFromWeightMap(weightMap): void

• Parameters

  • weightMap: NamedTensorMap

  Returns void

reassignParamFromPath

• reassignParamFromPath(paramPath, tensor): void

• Parameters

  • paramPath: string

  • tensor: Tensor<Rank>

  Returns void

serializeParams

• serializeParams(): Float32Array

• Returns Float32Array

variable

• variable(): void

• Returns void