Function listModels

  • listModels(): Promise<{
        [url: string]: ModelArtifactsInfo;
    }>

  • List all models stored in registered storage mediums.

    For a web browser environment, the registered mediums are Local Storage and IndexedDB.

    // First create and save a model.
    const model = tf.sequential();
    model.add(tf.layers.dense(
        {units: 1, inputShape: [10], activation: 'sigmoid'}));
    await model.save('localstorage://demo/management/model1');

    // Then list existing models.
    console.log(JSON.stringify(await tf.io.listModels()));

    // Delete the model.
    await tf.io.removeModel('localstorage://demo/management/model1');

    // List models again.
    console.log(JSON.stringify(await tf.io.listModels()));

    Returns Promise<{
        [url: string]: ModelArtifactsInfo;
    }>

    A Promise of a dictionary mapping URLs of existing models to their model artifacts info. URLs include medium-specific schemes, e.g., 'indexeddb://my/model/1'. Model artifacts info include type of the model's topology, byte sizes of the topology, weights, etc.

    Doc

Settings

Member Visibility