new version

SciML · Mar 3, 2020 · 4783915 · 4783915 · MartinuzziFrancesco · Mar 3, 2020
1 parent 833c32a
commit 4783915
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "ReservoirComputing"
 uuid = "7c2d2b1e-3dd4-11ea-355a-8f6a8116e294"
 authors = ["Francesco Martinuzzi"]
-version = "0.2.1"
+version = "0.3.0"
 
 [deps]
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"

diff --git a/README.md b/README.md
@@ -61,7 +61,8 @@ esn = ESN(approx_res_size,
     sigma, #default = 0.1
     alpha, #default = 1.0
     beta, #default = 0.0
-    nonlin_alg #default = "None")
+    nonlin_alg #default = "None"
+    )
 ```
 The echo state network can now be trained and tested:
 ```julia

diff --git a/src/echostatenetwork.jl b/src/echostatenetwork.jl
@@ -2,7 +2,7 @@ struct ESN{T<:AbstractFloat}
     res_size::Integer
     in_size::Integer
     out_size::Integer
-    train_data::Matrix{T}
+    train_data::Array{T}
     degree::Integer
     sigma::T
     alpha::T
@@ -15,7 +15,7 @@ struct ESN{T<:AbstractFloat}
     states::Matrix{T}
 
     function ESN(approx_res_size::Integer,
-            train_data::Matrix{T},
+            train_data::Array{T},
             degree::Integer,
             radius::T,
             activation::Function = tanh,
@@ -65,7 +65,7 @@ end
 
 function states_matrix(W::Matrix{Float64},
         W_in::Matrix{Float64},
-        train_data::Matrix{Float64},
+        train_data::Array{Float64},
         alpha::Float64,
         activation::Function)