salim4n/tfjs-dqn-test-1744563808991

TensorFlow.js Model

Model Information

• Framework: TensorFlow.js
• Type: Deep Q-Network (DQN)
• Created by: IgnitionAI

Model Format

This model is saved in TensorFlow.js format and can be loaded in two ways:

1. LayersModel (Default)

   • Better for fine-tuning and training
   • More flexible for model modifications
   • Higher memory usage
   • Slower inference

2. GraphModel

   • Optimized for inference only
   • Faster execution
   • Lower memory usage
   • Not suitable for training

Usage

import * as tf from '@tensorflow/tfjs';

// Option 1: Load as LayersModel (for training/fine-tuning)
const layersModel = await tf.loadLayersModel('https://huggingface.co/salim4n/tfjs-dqn-test-1744563808991/resolve/main/model/model.json');

// Option 2: Load as GraphModel (for inference only)
const graphModel = await tf.loadGraphModel('https://huggingface.co/salim4n/tfjs-dqn-test-1744563808991/resolve/main/model/model.json');

// Run inference
const input = tf.tensor2d([[0.1, 0.2]]);
const output = model.predict(input);

Files

• model.json: Model architecture and configuration
• weights.bin: Model weights
• README.md: This documentation

Repository

This model was uploaded via the IgnitionAI TensorFlow.js integration.