Class NeuralNetwork<TNetParams>Abstract

constructor

• new NeuralNetwork<TNetParams>(name: string): 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?: boolean): void

• Parameters

  • throwOnRedispose: boolean = true

  Returns void

extractWeights

• extractWeights(weights: Float32Array): void

• Parameters

  • weights: Float32Array

  Returns void

freeze

• freeze(): void

• Returns void

getFrozenParams

• getFrozenParams(): { path: string; tensor: tf.Tensor }[]

• Returns { path: string; tensor: tf.Tensor }[]

getParamFromPath

• getParamFromPath(paramPath: string): tf.Tensor

• Parameters

  • paramPath: string

  Returns tf.Tensor

getParamList

• getParamList(): { path: string; tensor: tf.Tensor }[]

• Returns { path: string; tensor: tf.Tensor }[]

getTrainableParams

• getTrainableParams(): { path: string; tensor: tf.Tensor }[]

• Returns { path: string; tensor: tf.Tensor }[]

load

• load(weightsOrUrl: undefined | string | Float32Array): Promise<void>

• Parameters

  • weightsOrUrl: undefined | string | Float32Array

  Returns Promise<void>

loadFromDisk

• loadFromDisk(filePath: undefined | string): Promise<void>

• Parameters

  • filePath: undefined | string

  Returns Promise<void>

loadFromUri

• loadFromUri(uri: undefined | string): Promise<void>

• Parameters

  • uri: undefined | string

  Returns Promise<void>

loadFromWeightMap

• loadFromWeightMap(weightMap: NamedTensorMap): void

• Parameters

  • weightMap: NamedTensorMap

  Returns void

reassignParamFromPath

• reassignParamFromPath(paramPath: string, tensor: tf.Tensor): void

• Parameters

  • paramPath: string
  • tensor: tf.Tensor

  Returns void

serializeParams

• serializeParams(): Float32Array

• Returns Float32Array

variable

• variable(): void

• Returns void