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