Hi, how can I save a nn I've trained up to disk, and load into NodeJ

If you have the nn params in a java object (i.e. the way params are passed aroun

thanks. I found this, which seems to help :) <a href="https://www.npmjs.com/pa

Persistence to disk about adnn HOT 2 CLOSED

treidio commented on July 28, 2024

Persistence to disk

from adnn.

Comments (2)

dritchie commented on July 28, 2024

If you have the nn params in a javascript object (i.e. the way params are passed around in the opt module methods), then you can pretty much just JSON.serialize that object. The only caveat is that you'll want to handle the Tensor objects a bit specially (so that they can be recognized as instances of the Tensor class when they're deserialized). Here's some code that I've used to do this in a project that uses adnn:

'use strict';

var _ = require('lodash')
var Tensor = require('../tensor');

function tensorToObject(tensor) {
  return {
    dims: tensor.dims,
    data: tensor.toFlatArray()
  };
}

function tensorsToObjects(paramObj) {
  var prms = _.mapValues(paramObj, function(lst) {
    return lst.map(tensorToObject);
  });
  return prms;
}

function objectsToTensors(paramObj) {
  var prms = {};
  for (var name in paramObj) {
    prms[name] = paramObj[name].map(function(tensor) {
      return new Tensor(tensor.dims).fromFlatArray(tensor.data);
    });
  }
  return prms;
}

function serializeParams(paramObj) {
  return JSON.stringify(tensorsToObjects(paramObj));
}

function deserializeParams(str) {
  return objectsToTensors(JSON.parse(str));
}

module.exports = {
  tensorsToObjects: tensorsToObjects,
  objectsToTensors: objectsToTensors,
  serializeParams: serializeParams,
  deserializeParams: deserializeParams
};