diff --git a/algovivo/nn/NeuralFramePolicy.js b/algovivo/nn/MLPPolicy.js similarity index 98% rename from algovivo/nn/NeuralFramePolicy.js rename to algovivo/nn/MLPPolicy.js index 419218d3..c78dddc6 100644 --- a/algovivo/nn/NeuralFramePolicy.js +++ b/algovivo/nn/MLPPolicy.js @@ -5,7 +5,7 @@ function sampleNormal(mean, stdDev) { return mean + z * stdDev; } -class NeuralFramePolicy { +class MLPPolicy { constructor(args = {}) { if (args.system == null) { throw new Error("system required to create policy"); @@ -128,4 +128,4 @@ class NeuralFramePolicy { } } -module.exports = NeuralFramePolicy; \ No newline at end of file +module.exports = MLPPolicy; \ No newline at end of file diff --git a/algovivo/nn/index.js b/algovivo/nn/index.js index 57c702a4..20080b30 100644 --- a/algovivo/nn/index.js +++ b/algovivo/nn/index.js @@ -1,3 +1,3 @@ module.exports = { - NeuralFramePolicy: require("./NeuralFramePolicy") + MLPPolicy: require("./MLPPolicy") } \ No newline at end of file diff --git a/demo/src/AgentSystem.js b/demo/src/AgentSystem.js index cacd4a45..6ce8456f 100644 --- a/demo/src/AgentSystem.js +++ b/demo/src/AgentSystem.js @@ -23,7 +23,7 @@ export default class AgentSystem { this.system.set(mesh); if (policy != null) { - this.policy = new this.algovivo.nn.NeuralFramePolicy({ system: this.system }); + this.policy = new this.algovivo.nn.MLPPolicy({ system: this.system }); this.policy.loadData(policy); } } diff --git a/test/nn/generateTrajectory.js b/test/nn/generateTrajectory.js index d9b2b0fc..5bdb6f61 100644 --- a/test/nn/generateTrajectory.js +++ b/test/nn/generateTrajectory.js @@ -31,7 +31,7 @@ async function main() { triangles: meshData.triangles, trianglesRsi: meshData.rsi }); - const policy = new algovivo.nn.NeuralFramePolicy({ + const policy = new algovivo.nn.MLPPolicy({ system: system, active: true }); diff --git a/test/nn/trajectory.test.js b/test/nn/trajectory.test.js index 7f727b9c..4f2b8508 100644 --- a/test/nn/trajectory.test.js +++ b/test/nn/trajectory.test.js @@ -34,7 +34,7 @@ test("neural frame policy", async () => { }); expect(system.l0.toArray()).toEqual(meshData.l0); expect(system.rsi.toArray()).toEqual(meshData.rsi); - const policy = new algovivo.nn.NeuralFramePolicy({ + const policy = new algovivo.nn.MLPPolicy({ system: system, stochastic: false, active: true