@@ -10,7 +10,7 @@ struct RProp_params
1010end
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)
1414Trains kriging hyperparameters with RProp.
1515"""
1616function 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)
5656end
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)
6060This trains a kriging model by using maximum likelihood with stochastic gradient descent.
6161"""
6262function 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