1
0
Fork 0
mirror of https://github.com/Findus23/nn_evaluate.git synced 2024-09-19 14:53:44 +02:00
nn_evaluate/typescript/data.ts

21 lines
642 B
TypeScript

import data from "../pytorch_model.json"
export const means: number[] = data.means
export const stds: number[] = data.stds
export const hiddenWeight: number[][] = data["hidden.weight"]
export const hiddenBias: number[] = data["hidden.bias"]
export const outputWeight: number[][] = data["output.weight"]
export const outputBias: number[] = data["output.bias"]
export const hiddenLayerSize = hiddenBias.length
export const inputLayerSize = means.length
export const outputLayerSize = outputBias.length
const ang = 30
const v = 2
const m = 1e24
const gamma = 0.6
const wp = 1e-4
const wt = wp
export const input = [ang, v, m, gamma, wt, wp]