Views
No views yet
1import { loadModelFromHub } from '@ignitionai/backend-tfjs';
2
3// Option 1: Load as LayersModel (for training/fine-tuning)
4const layersModel = await loadModelFromHub(
5 'salim4n/tfjs-dqn-test-1744563833356',
6 'model/model.json',
7 false // graphModel = false for LayersModel
8);
9
10// Option 2: Load as GraphModel (for inference only)
11const graphModel = await loadModelFromHub(
12 'salim4n/tfjs-dqn-test-1744563833356',
13 'model/model.json',
14 true // graphModel = true for GraphModel
15);
16
17// Run inference
18const input = tf.tensor2d([[0.1, 0.2]]);
19const output = model.predict(input);model.json: Model architecture and configurationweights.bin: Model weightsREADME.md: This documentation