Skip to content

Commit

Permalink
Fix composition weights schema
Browse files Browse the repository at this point in the history
  • Loading branch information
PicoCentauri committed Jul 24, 2024
1 parent e14207b commit 4c42e3d
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 2 deletions.
1 change: 1 addition & 0 deletions docs/src/architectures/soap-bpnn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -147,6 +147,7 @@ The parameters for training are
:param learning_rate: learning rate
:param log_interval: number of epochs that elapse between reporting new training results
:param checkpoint_interval: Interval to save a checkpoint to disk.
:param fixed_composition_weights: TODO including an example.
:param per_atom_targets: Specifies whether the model should be trained on a per-atom
loss. In that case, the logger will also output per-atom metrics for that target. In
any case, the final summary will be per-structure.
Expand Down
10 changes: 8 additions & 2 deletions src/metatrain/experimental/soap_bpnn/schema-hypers.json
Original file line number Diff line number Diff line change
Expand Up @@ -120,8 +120,14 @@
"fixed_composition_weights": {
"type": "object",
"patternProperties": {
"^[0-9]+$": {
"type": "number"
"^.*$": {
"type": "object",
"patternProperties": {
"^[0-9]+$": {
"type": "number"
}
},
"additionalProperties": false
}
},
"additionalProperties": false
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -195,3 +195,7 @@ def test_output_per_atom():

assert outputs["energy"].block().samples.names == ["system", "atom"]
assert outputs["energy"].block().values.shape == (4, 1)


def test_fixed_composition_weights():
pass

0 comments on commit 4c42e3d

Please sign in to comment.