some fixes

SciML · Dec 23, 2022 · 0a9fe2e · 0a9fe2e
1 parent 7109779
commit 0a9fe2e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/src/ReservoirComputing.jl b/src/ReservoirComputing.jl
@@ -19,7 +19,7 @@ export AbstractLayer, create_layer
 export WeightedLayer, DenseLayer, SparseLayer, MinimumLayer, InformedLayer, NullLayer
 export BernoulliSample, IrrationalSample
 export GaussianProcess
-export AbstractReservoir, create_reservoir
+export AbstractReservoir, create_reservoir, create_states
 export RandSparseReservoir, PseudoSVDReservoir, DelayLineReservoir
 export DelayLineBackwardReservoir, SimpleCycleReservoir, CycleJumpsReservoir, NullReservoir
 export RNN, MRNN, GRU, GRUParams, FullyGated, Variant1, Variant2, Variant3, Minimal

diff --git a/src/esn/esn_reservoir_drivers.jl b/src/esn/esn_reservoir_drivers.jl
@@ -1,9 +1,12 @@
 abstract type AbstractReservoirDriver end
 
 """
-    create_states(
-        reservoir_driver::AbstractReservoirDriver, train_data, reservoir_matrix,input_matrix
-    )
+    create_states(reservoir_driver::AbstractReservoirDriver,
+                       train_data,
+                       washout,
+                       reservoir_matrix,
+                       input_matrix,
+                       bias_vector)
 
 Return the trained ESN states according to the given driver.
 """