diff options
| author | Your Name <[email protected]> | 2026-07-15 14:41:00 -0700 |
|---|---|---|
| committer | Your Name <[email protected]> | 2026-07-15 14:43:00 -0700 |
| commit | cc587b9aa209137c68e888aed50a8ee18f5bfaee (patch) | |
| tree | cfd2dca4cb1650eebf2e4046cea5ce9f9280e606 /src/torch_reservoir.py | |
| parent | 99c240a1c69da19813c50f167cd51456208479d2 (diff) | |
functional SPICE crossbar
Diffstat (limited to 'src/torch_reservoir.py')
| -rw-r--r-- | src/torch_reservoir.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/torch_reservoir.py b/src/torch_reservoir.py index 6e7ceb7..a21ed75 100644 --- a/src/torch_reservoir.py +++ b/src/torch_reservoir.py @@ -14,7 +14,7 @@ import time # ----- PARAMETERS ----- NUM_INPUTS = 1 NUM_OUTPUTS = 1 -NUM_NEURONS = 2000 +NUM_NEURONS = 800 time_steps = 2000 input_layer = torch.nn.Linear(NUM_INPUTS, NUM_NEURONS, bias=False) @@ -122,12 +122,12 @@ test_steps = np.arange(800, time_steps) plt.figure(figsize=(10, 5)) plt.plot(test_steps, y_test, label="True Future Wave", color="black", linestyle="dashed") plt.plot(test_steps, predictions, label="Reservoir Prediction", color="blue", alpha=0.8) -plt.title("spires memristor Time-Series Forecasting") +plt.title("spires memristor Time-Series prediction") plt.xlabel("time steps") plt.ylabel("Amplitude") plt.legend() plt.tight_layout() -plt.show() +plt.savefig("memristor_spires_proof.png") free_spires_reservoir(spires_reservoir) |