Create neuralNetwork.js

KOSASIH · Jul 21, 2024 · a4c9903 · a4c9903
1 parent 4b4a2a3
commit a4c9903
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/utils/neuralNetwork.js b/utils/neuralNetwork.js
@@ -0,0 +1,16 @@
+// neuralNetwork.js
+import * as NeuralNetwork from 'neural-network';
+
+const neuralNetworkUtils = {
+  createNeuralNetwork: (architecture) => {
+    return new NeuralNetwork.NeuralNetwork(architecture);
+  },
+  trainNeuralNetwork: (neuralNetwork, dataset) => {
+    return neuralNetwork.train(dataset);
+  },
+  predictWithNeuralNetwork: (neuralNetwork, input) => {
+    return neuralNetwork.predict(input);
+  },
+};
+
+export default neuralNetworkUtils;