Skip to content

Commit

Permalink
Added some comments
Browse files Browse the repository at this point in the history
  • Loading branch information
stevetorr committed Oct 25, 2019
1 parent 6ccaf6b commit c010e0a
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions flare/gp_from_aimd.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,7 @@ def __init__(self, frames: List[Structure],
noise variance hyperparameter
:param abs_std_tolerance: Train if uncertainty is above this
:param parallel: Use parallel functions or not
:param validate_ratio: Fraction of frames used for validation
:param no_cpus: number of cpus to run with multithreading
:param skip: Skip through frames
:param calculate_energy: Use local energy kernel or not
Expand Down Expand Up @@ -168,6 +169,7 @@ def pre_run(self):
# so all atomic species are represented in the first step.
# Otherwise use the seed frames passed in by user.

# Remove frames used as seed from later part of training
if self.pre_train_on_skips > 0:
self.seed_frames = []
newframes = []
Expand All @@ -177,6 +179,7 @@ def pre_run(self):
else:
newframes += [self.frames[i]]
self.frames = newframes

elif len(self.gp.training_data) == 0 and len(self.seed_frames) == 0:
self.seed_frames = [self.frames[0]]
self.frames = self.frames[1:]
Expand Down

0 comments on commit c010e0a

Please sign in to comment.