Skip to content

Commit

Permalink
Small fizxes in documentation examples.
Browse files Browse the repository at this point in the history
  • Loading branch information
emmanuellujan committed Jul 14, 2024
1 parent 3942641 commit 657f948
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions examples/DPP-ACE-Na/fit-dpp-ace-na.jl
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ using LinearAlgebra, Plots

# Define paths.
base_path = haskey(ENV, "BASE_PATH") ? ENV["BASE_PATH"] : "../../"
ds_path = "$base_path/examples/data/Na/liquify_sodium.yaml"
ds_path = "$base_path/examples/data/Na/liquify_sodium.yaml";

# ## Load datasets

Expand Down Expand Up @@ -44,7 +44,7 @@ ds_train = DataSet(conf_train .+ e_descr_train);
dpp = kDPP(ds_train, GlobalMean(), DotProduct(); batch_size = 200)

# Subsample trainig dataset.
dpp_inds = get_random_subset(dpp)
dpp_inds = get_random_subset(dpp);

# ## Learn coefficients

Expand All @@ -57,7 +57,7 @@ lb = LBasisPotential(ace)

# Update test dataset by adding energy descriptors.
println("Computing local descriptors of test dataset")
e_descr_test = compute_local_descriptors(conf_test, ace)
e_descr_test = compute_local_descriptors(conf_test, ace, pbar = false)
ds_test = DataSet(conf_test .+ e_descr_test);

# Get true and predicted energy values (assuming that all configurations have the same no. of atoms).
Expand Down
2 changes: 1 addition & 1 deletion examples/LJ-Ar/lennard-jones-ar.jl
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ using LinearAlgebra, Plots, DisplayAs

# Define paths.
base_path = haskey(ENV, "BASE_PATH") ? ENV["BASE_PATH"] : "../../"
ds_path = "$base_path/examples/data/LJ-AR/lj-ar.yaml"
ds_path = "$base_path/examples/data/LJ-AR/lj-ar.yaml";

# ## Load datasets
ds, thermo = load_data(ds_path, YAML(:Ar, u"eV", u""))
Expand Down

0 comments on commit 657f948

Please sign in to comment.