Skip to content

Commit

Permalink
minro edit while HGF package is updated
Browse files Browse the repository at this point in the history
  • Loading branch information
PTWaade committed Apr 14, 2024
1 parent ee370b6 commit 61848c5
Showing 1 changed file with 15 additions and 15 deletions.
30 changes: 15 additions & 15 deletions test/testsuite/HGF_tests.jl
Original file line number Diff line number Diff line change
@@ -1,21 +1,21 @@
using ActionModels
using HierarchicalGaussianFiltering
using Distributions
using Plots
using StatsPlots
# using ActionModels
# using HierarchicalGaussianFiltering
# using Distributions
# using Plots
# using StatsPlots

agent = premade_agent(
"hgf_gaussian",
premade_hgf("continuous_2level", verbose = false),
verbose = false,
)
# agent = premade_agent(
# "hgf_gaussian_action",
# premade_hgf("continuous_2level", verbose = false),
# verbose = false,
# )

priors = Dict(("x1", "volatility") => Normal(-5, 1))
# priors = Dict(("x1", "volatility") => Normal(-5, 1))

inputs = [1, 1.2, 1.4]
# inputs = [1, 1.2, 1.4]

actions = [1, 1.1, 1.5]
# actions = [1, 1.1, 1.5]

results = fit_model(agent, priors, inputs, actions, n_iterations = 100, n_chains = 1)
# results = fit_model(agent, priors, inputs, actions, n_iterations = 100, n_chains = 1)

plot_parameter_distribution(results, priors)
# plot_parameter_distribution(results, priors)

0 comments on commit 61848c5

Please sign in to comment.