@ignitionai/backend-tfjs
Version:
TensorFlow.js backend for IgnitionAI - browser-based reinforcement learning framework
11 lines (10 loc) • 465 B
TypeScript
import * as tf from '@tensorflow/tfjs-node';
/**
* Save a TensorFlow.js model locally and push it to Hugging Face Hub.
*
* @param model Trained tf.LayersModel
* @param repo Full Hugging Face repo ID (e.g. "salim4n/dqn-agent")
* @param token Hugging Face access token
* @param subdir Subfolder inside repo (e.g. "step-5" or "best")
*/
export declare function saveModelToHub(model: tf.LayersModel, repo: string, token: string, subdir?: string): Promise<void>;