saving and loading in max dicts
This commit is contained in:
trian-gles
parent
847db766d2
commit
43cd89faa5
40
tf.js
40
tf.js
@ -8,7 +8,7 @@ const outputShape = parseInt(args[1]);
|
|||||||
const hiddenSize = parseInt(args[2]);
|
const hiddenSize = parseInt(args[2]);
|
||||||
|
|
||||||
// Define a model for linear regression.
|
// Define a model for linear regression.
|
||||||
const model = tf.sequential();
|
var model = tf.sequential();
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@ -47,3 +47,41 @@ maxApi.addHandler("predict", (...data) => {
|
|||||||
|
|
||||||
});
|
});
|
||||||
|
|
||||||
|
async function getWeights() {
|
||||||
|
let weights = model.getWeights();
|
||||||
|
for (let i = 0; i < weights.length; i++){
|
||||||
|
let data = await weights[i].data();
|
||||||
|
let shape = weights[i].shape;
|
||||||
|
weights[i] = {data, shape};
|
||||||
|
|
||||||
|
}
|
||||||
|
return weights;
|
||||||
|
}
|
||||||
|
|
||||||
|
maxApi.addHandler("save", (dictId, key) => {
|
||||||
|
maxApi.getDict(dictId).then((dict) => {
|
||||||
|
getWeights().then((weights) => {
|
||||||
|
console.log(weights);
|
||||||
|
dict[key] = weights;
|
||||||
|
maxApi.setDict(dictId, dict);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
maxApi.addHandler("load", (dictId, key) => {
|
||||||
|
maxApi.getDict(dictId).then((dict) => {
|
||||||
|
let data = dict[key];
|
||||||
|
let tensors = [];
|
||||||
|
data.forEach(item => {
|
||||||
|
let shape = item.shape;
|
||||||
|
let vals = [];
|
||||||
|
for (const [key, value] of Object.entries(item.data)) {
|
||||||
|
vals.push(value);
|
||||||
|
|
||||||
|
}
|
||||||
|
tensors.push(tf.tensor(vals, shape));
|
||||||
|
});
|
||||||
|
model.setWeights(tensors);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
|||||||
151
tf.maxpat
151
tf.maxpat
@ -10,7 +10,7 @@
|
|||||||
}
|
}
|
||||||
,
|
,
|
||||||
"classnamespace" : "box",
|
"classnamespace" : "box",
|
||||||
"rect" : [ 1649.0, 182.0, 948.0, 575.0 ],
|
"rect" : [ 164.0, 165.0, 948.0, 575.0 ],
|
||||||
"bglocked" : 0,
|
"bglocked" : 0,
|
||||||
"openinpresentation" : 0,
|
"openinpresentation" : 0,
|
||||||
"default_fontsize" : 12.0,
|
"default_fontsize" : 12.0,
|
||||||
@ -39,6 +39,117 @@
|
|||||||
"subpatcher_template" : "",
|
"subpatcher_template" : "",
|
||||||
"assistshowspatchername" : 0,
|
"assistshowspatchername" : 0,
|
||||||
"boxes" : [ {
|
"boxes" : [ {
|
||||||
|
"box" : {
|
||||||
|
"id" : "obj-40",
|
||||||
|
"linecount" : 2,
|
||||||
|
"maxclass" : "message",
|
||||||
|
"numinlets" : 2,
|
||||||
|
"numoutlets" : 1,
|
||||||
|
"outlettype" : [ "" ],
|
||||||
|
"patching_rect" : [ 237.0, 453.0, 50.0, 36.0 ],
|
||||||
|
"text" : "9. 10. 11."
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
, {
|
||||||
|
"box" : {
|
||||||
|
"id" : "obj-36",
|
||||||
|
"maxclass" : "message",
|
||||||
|
"numinlets" : 2,
|
||||||
|
"numoutlets" : 1,
|
||||||
|
"outlettype" : [ "" ],
|
||||||
|
"patching_rect" : [ 768.0, 156.0, 111.0, 22.0 ],
|
||||||
|
"text" : "load weights kieran"
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
, {
|
||||||
|
"box" : {
|
||||||
|
"id" : "obj-35",
|
||||||
|
"maxclass" : "message",
|
||||||
|
"numinlets" : 2,
|
||||||
|
"numoutlets" : 1,
|
||||||
|
"outlettype" : [ "" ],
|
||||||
|
"patching_rect" : [ 626.0, 156.0, 114.0, 22.0 ],
|
||||||
|
"text" : "save weights kieran"
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
, {
|
||||||
|
"box" : {
|
||||||
|
"data" : {
|
||||||
|
"kieran" : [ {
|
||||||
|
"data" : {
|
||||||
|
"0" : 0.927409112453461,
|
||||||
|
"1" : -0.10581861436367,
|
||||||
|
"2" : 0.110250025987625,
|
||||||
|
"3" : -1.001458287239075,
|
||||||
|
"4" : 0.689840376377106,
|
||||||
|
"5" : 0.396656483411789,
|
||||||
|
"6" : -1.090744495391846,
|
||||||
|
"7" : -0.033253565430641
|
||||||
|
}
|
||||||
|
,
|
||||||
|
"shape" : [ 2, 4 ]
|
||||||
|
}
|
||||||
|
, {
|
||||||
|
"data" : {
|
||||||
|
"0" : -0.04672335088253,
|
||||||
|
"1" : 0.441919326782227,
|
||||||
|
"2" : -0.127905443310738,
|
||||||
|
"3" : 0.395313948392868
|
||||||
|
}
|
||||||
|
,
|
||||||
|
"shape" : [ 4 ]
|
||||||
|
}
|
||||||
|
, {
|
||||||
|
"data" : {
|
||||||
|
"0" : 0.714273393154144,
|
||||||
|
"1" : 0.45893207192421,
|
||||||
|
"2" : 0.772015511989594,
|
||||||
|
"3" : -0.403766572475433,
|
||||||
|
"4" : 0.378416150808334,
|
||||||
|
"5" : 0.486913919448853,
|
||||||
|
"6" : -0.327280908823013,
|
||||||
|
"7" : -0.424903750419617,
|
||||||
|
"8" : -0.59527575969696,
|
||||||
|
"9" : -0.619619905948639,
|
||||||
|
"10" : -0.706632375717163,
|
||||||
|
"11" : -0.025489492341876
|
||||||
|
}
|
||||||
|
,
|
||||||
|
"shape" : [ 4, 3 ]
|
||||||
|
}
|
||||||
|
, {
|
||||||
|
"data" : {
|
||||||
|
"0" : -0.584373593330383,
|
||||||
|
"1" : 0.079081602394581,
|
||||||
|
"2" : 0.754254102706909
|
||||||
|
}
|
||||||
|
,
|
||||||
|
"shape" : [ 3 ]
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
,
|
||||||
|
"id" : "obj-33",
|
||||||
|
"maxclass" : "newobj",
|
||||||
|
"numinlets" : 2,
|
||||||
|
"numoutlets" : 5,
|
||||||
|
"outlettype" : [ "dictionary", "", "", "", "" ],
|
||||||
|
"patching_rect" : [ 768.0, 215.0, 133.0, 22.0 ],
|
||||||
|
"saved_object_attributes" : {
|
||||||
|
"embed" : 1,
|
||||||
|
"legacy" : 0,
|
||||||
|
"parameter_enable" : 0,
|
||||||
|
"parameter_mappable" : 0
|
||||||
|
}
|
||||||
|
,
|
||||||
|
"text" : "dict weights @embed 1"
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
, {
|
||||||
"box" : {
|
"box" : {
|
||||||
"id" : "obj-32",
|
"id" : "obj-32",
|
||||||
"maxclass" : "comment",
|
"maxclass" : "comment",
|
||||||
@ -68,7 +179,7 @@
|
|||||||
"maxclass" : "comment",
|
"maxclass" : "comment",
|
||||||
"numinlets" : 1,
|
"numinlets" : 1,
|
||||||
"numoutlets" : 0,
|
"numoutlets" : 0,
|
||||||
"patching_rect" : [ 339.0, 44.0, 150.0, 34.0 ],
|
"patching_rect" : [ 339.0, 44.0, 152.0, 34.0 ],
|
||||||
"text" : "Uses basic SGD optimizer with 0.01 learning rate"
|
"text" : "Uses basic SGD optimizer with 0.01 learning rate"
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -150,7 +261,6 @@
|
|||||||
"numoutlets" : 1,
|
"numoutlets" : 1,
|
||||||
"outlettype" : [ "" ],
|
"outlettype" : [ "" ],
|
||||||
"patching_rect" : [ 262.0, 163.0, 55.0, 22.0 ],
|
"patching_rect" : [ 262.0, 163.0, 55.0, 22.0 ],
|
||||||
"presentation_linecount" : 3,
|
|
||||||
"text" : "3 3 5 6 7"
|
"text" : "3 3 5 6 7"
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -163,7 +273,6 @@
|
|||||||
"numoutlets" : 1,
|
"numoutlets" : 1,
|
||||||
"outlettype" : [ "" ],
|
"outlettype" : [ "" ],
|
||||||
"patching_rect" : [ 330.0, 163.0, 55.0, 22.0 ],
|
"patching_rect" : [ 330.0, 163.0, 55.0, 22.0 ],
|
||||||
"presentation_linecount" : 3,
|
|
||||||
"text" : "4 4 7 8 9"
|
"text" : "4 4 7 8 9"
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -176,7 +285,6 @@
|
|||||||
"numoutlets" : 1,
|
"numoutlets" : 1,
|
||||||
"outlettype" : [ "" ],
|
"outlettype" : [ "" ],
|
||||||
"patching_rect" : [ 197.0, 163.0, 55.0, 22.0 ],
|
"patching_rect" : [ 197.0, 163.0, 55.0, 22.0 ],
|
||||||
"presentation_linecount" : 3,
|
|
||||||
"text" : "2 2 3 4 5"
|
"text" : "2 2 3 4 5"
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -192,19 +300,6 @@
|
|||||||
"text" : "1 1 1 2 3"
|
"text" : "1 1 1 2 3"
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
|
||||||
, {
|
|
||||||
"box" : {
|
|
||||||
"id" : "obj-11",
|
|
||||||
"linecount" : 6,
|
|
||||||
"maxclass" : "message",
|
|
||||||
"numinlets" : 2,
|
|
||||||
"numoutlets" : 1,
|
|
||||||
"outlettype" : [ "" ],
|
|
||||||
"patching_rect" : [ 192.0, 413.0, 50.0, 91.0 ],
|
|
||||||
"text" : "8.914521 10.032911 11.035411"
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
}
|
||||||
, {
|
, {
|
||||||
"box" : {
|
"box" : {
|
||||||
@ -246,7 +341,7 @@
|
|||||||
"numoutlets" : 1,
|
"numoutlets" : 1,
|
||||||
"offset" : [ 0.0, 0.0 ],
|
"offset" : [ 0.0, 0.0 ],
|
||||||
"outlettype" : [ "bang" ],
|
"outlettype" : [ "bang" ],
|
||||||
"patching_rect" : [ 484.0, 353.0, 400.0, 220.0 ],
|
"patching_rect" : [ 526.0, 284.0, 400.0, 220.0 ],
|
||||||
"viewvisibility" : 1
|
"viewvisibility" : 1
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -275,8 +370,6 @@
|
|||||||
"args" : [ 2, 3, 4 ],
|
"args" : [ 2, 3, 4 ],
|
||||||
"autostart" : 1,
|
"autostart" : 1,
|
||||||
"defer" : 0,
|
"defer" : 0,
|
||||||
"node_bin_path" : "",
|
|
||||||
"npm_bin_path" : "",
|
|
||||||
"watch" : 1
|
"watch" : 1
|
||||||
}
|
}
|
||||||
,
|
,
|
||||||
@ -287,7 +380,7 @@
|
|||||||
],
|
],
|
||||||
"lines" : [ {
|
"lines" : [ {
|
||||||
"patchline" : {
|
"patchline" : {
|
||||||
"destination" : [ "obj-11", 1 ],
|
"destination" : [ "obj-40", 1 ],
|
||||||
"source" : [ "obj-1", 0 ]
|
"source" : [ "obj-1", 0 ]
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -368,6 +461,20 @@
|
|||||||
"source" : [ "obj-20", 0 ]
|
"source" : [ "obj-20", 0 ]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
, {
|
||||||
|
"patchline" : {
|
||||||
|
"destination" : [ "obj-1", 0 ],
|
||||||
|
"source" : [ "obj-35", 0 ]
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
, {
|
||||||
|
"patchline" : {
|
||||||
|
"destination" : [ "obj-1", 0 ],
|
||||||
|
"source" : [ "obj-36", 0 ]
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
, {
|
, {
|
||||||
"patchline" : {
|
"patchline" : {
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user