77 lines
2.8 KiB
JavaScript
77 lines
2.8 KiB
JavaScript
|
// ShapeSeq — Sequence NISPS Instance
|
||
|
|
//
|
||
|
|
// Factory for creating a second WasmIML instance dedicated to sequence control.
|
||
|
|
// The MLP has a fixed 16-element output which a downstream param mapping layer
|
||
|
|
// (not in this module) fans out to however many primitive params the current
|
||
|
|
// chain requires.
|
||
|
|
//
|
||
|
|
// Input routing is configurable and wired externally (default: hand tracking
|
||
|
|
// features 0+1). This module only creates the IML instance — it does not
|
||
|
|
// subscribe to any input source. The integration layer (see meml-9tf) is
|
||
|
|
// responsible for calling setInputs() with routed values each frame.
|
||
|
|
//
|
||
|
|
// Usage:
|
||
|
|
// import { createSequenceIML } from './shapeseq/seq-iml.js';
|
||
|
|
// const seqIML = await createSequenceIML();
|
||
|
|
// // Per frame:
|
||
|
|
// seqIML.setInputs([x, y]);
|
||
|
|
// seqIML.process();
|
||
|
|
// const params16 = seqIML.getOutputs();
|
||
|
|
|
||
|
|
import { WasmIML } from '../nisps/nisps-wasm.js';
|
||
|
|
|
||
|
|
// Fixed architecture for the sequence MLP.
|
||
|
|
// 2 inputs (routed externally), 16 outputs (fixed — param mapping fans out).
|
||
|
|
// Smaller network than the synth MLP ([3,32,48,64,126]):
|
||
|
|
// three hidden layers of 16 neurons keeps inference cheap while providing
|
||
|
|
// enough capacity for 16 continuous outputs.
|
||
|
|
const SEQ_N_INPUTS = 2;
|
||
|
|
const SEQ_N_OUTPUTS = 16;
|
||
|
|
const SEQ_HIDDEN_LAYERS = [16, 16, 16];
|
||
|
|
|
||
|
|
// Training hyperparameters — same defaults as the synth IML
|
||
|
|
const SEQ_MAX_ITERATIONS = 1000;
|
||
|
|
const SEQ_LEARNING_RATE = 1.0;
|
||
|
|
const SEQ_CONVERGENCE_THRESHOLD = 0.00001;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Create a WasmIML instance configured for sequence control.
|
||
|
|
*
|
||
|
|
* Follows the same creation pattern as imlJoy / imlHand in a-app.js:
|
||
|
|
* const iml = await WasmIML.create(nInputs, nOutputs, hiddenLayers, ...);
|
||
|
|
*
|
||
|
|
* The returned object is a standard WasmIML instance with the full interface:
|
||
|
|
* setInput / setInputs, getOutputs, process,
|
||
|
|
* addExample, clearDataset, train, trainAsync,
|
||
|
|
* randomiseWeights (drawWeights), moveWeights,
|
||
|
|
* exampleCount, destroy, etc.
|
||
|
|
*
|
||
|
|
* Additionally exposes SEQ_N_INPUTS, SEQ_N_OUTPUTS, SEQ_HIDDEN_LAYERS as
|
||
|
|
* properties on the returned object for introspection by downstream code.
|
||
|
|
*
|
||
|
|
* @returns {Promise<WasmIML>} — the sequence IML instance (augmented with
|
||
|
|
* .SEQ_N_INPUTS, .SEQ_N_OUTPUTS, .SEQ_HIDDEN_LAYERS)
|
||
|
|
*/
|
||
|
|
export async function createSequenceIML() {
|
||
|
|
const seqIML = await WasmIML.create(
|
||
|
|
SEQ_N_INPUTS,
|
||
|
|
SEQ_N_OUTPUTS,
|
||
|
|
SEQ_HIDDEN_LAYERS,
|
||
|
|
SEQ_MAX_ITERATIONS,
|
||
|
|
SEQ_LEARNING_RATE,
|
||
|
|
SEQ_CONVERGENCE_THRESHOLD
|
||
|
|
);
|
||
|
|
|
||
|
|
seqIML.setLogger(msg => console.log('[NISPS:seq]', msg));
|
||
|
|
|
||
|
|
// Attach architecture metadata for introspection
|
||
|
|
seqIML.SEQ_N_INPUTS = SEQ_N_INPUTS;
|
||
|
|
seqIML.SEQ_N_OUTPUTS = SEQ_N_OUTPUTS;
|
||
|
|
seqIML.SEQ_HIDDEN_LAYERS = SEQ_HIDDEN_LAYERS;
|
||
|
|
|
||
|
|
return seqIML;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Re-export constants for use by other modules (e.g., param mapping layer)
|
||
|
|
export { SEQ_N_INPUTS, SEQ_N_OUTPUTS, SEQ_HIDDEN_LAYERS };
|