UNPKG

@ignitionai/backend-tfjs

Version:

TensorFlow.js backend for IgnitionAI - browser-based reinforcement learning framework

11 lines (10 loc) 465 B
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>;