Skip to content

Commit 84a24b7

Browse files
committed
fixing documentation
1 parent 039f01a commit 84a24b7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/calibrate_weights.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ struct RProp_params
1010
end
1111

1212
"""
13-
train_with_RProp(X::Array{Float64,2}, y::Array{Float64,1}, w_0::Float64, w_i::Array{Float64,1}, MaxIter::Int, noise::Float64, params::RProp_params)
13+
train_with_RProp(X::Array{Float64,2}, y::Array{Float64,1}, cov_func_parameters::gaussian_kernel_hyperparameters, MaxIter::Int, noise::Float64, params::RProp_params)
1414
Trains kriging hyperparameters with RProp.
1515
"""
1616
function calibrate_by_ML_with_Rprop(X::Array{Float64,2}, y::Array{Float64,1}, cov_func_parameters::gaussian_kernel_hyperparameters, MaxIter::Int, noise::Float64, params::RProp_params)
@@ -56,7 +56,7 @@ function sample(dim::Int, batch_size::Int)
5656
end
5757

5858
"""
59-
calibrate_by_ML_with_SGD(X::Array{Float64,2}, y::Array{Float64,1}, w_0::Float64, w_i::Array{Float64,1}, steps::Int, batch_size::Int, step_multiple::Float64, noise::Float64, seed::Int = 1988)
59+
calibrate_by_ML_with_SGD(X::Array{Float64,2}, y::Array{Float64,1}, cov_func_parameters::gaussian_kernel_hyperparameters, steps::Int, batch_size::Int, step_multiple::Float64, noise::Float64, seed::Int = 1988)
6060
This trains a kriging model by using maximum likelihood with stochastic gradient descent.
6161
"""
6262
function calibrate_by_ML_with_SGD(X::Array{Float64,2}, y::Array{Float64,1}, cov_func_parameters::gaussian_kernel_hyperparameters, steps::Int, batch_size::Int, step_multiple::Float64, noise::Float64, seed::Int = 1988)

0 commit comments

Comments
 (0)