Class NeuralNetwork<TNetParams>Abstract
Constructors
constructor
- new NeuralNetwork<TNetParams>(name: string): NeuralNetwork<TNetParams>
-
Accessors
isLoaded
- get isLoaded(): boolean
-
Returns boolean
paramMappings
- get paramMappings(): ParamMapping[]
-
Returns ParamMapping[]
params
- get params(): undefined | TNetParams
-
Returns undefined | TNetParams
Methods
dispose
- dispose(throwOnRedispose?: boolean): void
-
Parameters
-
throwOnRedispose: boolean = true
Returns void
freeze
- freeze(): void
-
Returns void
getFrozenParams
- getFrozenParams(): {
path: string;
tensor: Tensor;
}[]
-
Returns {
path: string;
tensor: Tensor;
}[]
getParamFromPath
- getParamFromPath(paramPath: string): Tensor
-
Returns Tensor
getParamList
- getParamList(): {
path: string;
tensor: Tensor;
}[]
-
Returns {
path: string;
tensor: Tensor;
}[]
getTrainableParams
- getTrainableParams(): {
path: string;
tensor: Tensor;
}[]
-
Returns {
path: string;
tensor: 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>
-
Returns Promise<void>
loadFromWeightMap
- loadFromWeightMap(weightMap: NamedTensorMap): void
-
Parameters
-
weightMap: NamedTensorMap
Returns void
reassignParamFromPath
- reassignParamFromPath(paramPath: string, tensor: Tensor): void
-
Parameters
-
paramPath: string
-
tensor: Tensor
Returns void
serializeParams
- serializeParams(): Float32Array
-
Returns Float32Array
variable
- variable(): void
-
Returns void