0.0.1 • Published 4 years ago

tfjs-clone v0.0.1

Weekly downloads
3
License
MIT
Repository
github
Last release
4 years ago

tfjs-clone

Creates a deep clone of an existing @tensorflow/tfjs model.

🚀 Getting Started

Using npm:

npm install --save tfjs-clone

Using yarn:

yarn add tfjs-clone

✍️ Usage

import * as tf from '@tensorflow/tfjs';
import { model as clone } from 'tfjs-clone';

const m = tf.sequential();

m.add(tf.layers.dense({ inputShape: [1], units: 16, activation: 'relu' }));
m.add(tf.layers.dense({ units: 1, activation: 'sigmoid' }));
m.compile({ optimizer: tf.train.rmsprop(1e-2), loss: 'binaryCrossentropy' });
  
const m2 = await clone(m);

✌️ License

MIT