atomistic-machine-learning / schnetpack

SchNetPack - Deep Neural Networks for Atomistic Systems
Other
791 stars 215 forks source link

Ep/ensemble calc #537

Open jnsLs opened 1 year ago

jnsLs commented 1 year ago

replaced ethanol model:

now implemented: SpkCalculator SpkEnsembleCalculator <- inherits from SpkCalculator BatchwiseCalculator <- inherits from SpkCalculator BatchwiseEnsembleCalculator <- inherits from SpkEnsembleCalculator SimpleEnsembleAverage (example for a particular average strategy)

the calculators have two new methods (to allow for more modular classes): _collect_results <-- calculates model outputs and converts property units _ase_specific_output_format <-- for individual structures this converts energy output to scalar value, for batches of structures it keeps the output in array form

updated batchwise optimization howto jupyter notebook --> 01_howto_batchwise_relaxations.ipynb

added howto_ensemble_calculations jupyter notebook --> 02_howto_ensemble_calculator