-
Notifications
You must be signed in to change notification settings - Fork 10
/
objective.py
90 lines (64 loc) · 3.31 KB
/
objective.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
"""Defines objective functions used in DFTFIT
"""
import math
import itertools
import numpy as np
# dft vs md comparison
def force_objective_function(md_calculations, dft_calculations):
n_force_sq_error = 0.0
d_force_sq_error = 0.0
for md_calculation, dft_calculation in zip(md_calculations, dft_calculations):
n_force_sq_error += np.sum((md_calculation.forces - dft_calculation.forces)**2.0)
d_force_sq_error += np.sum(dft_calculation.forces**2.0)
return math.sqrt(n_force_sq_error / d_force_sq_error)
def stress_objective_function(md_calculations, dft_calculations):
n_stress_sq_error = 0.0
d_stress_sq_error = 0.0
for md_calculation, dft_calculation in zip(md_calculations, dft_calculations):
n_stress_sq_error += np.sum((md_calculation.stress - dft_calculation.stress)**2.0)
d_stress_sq_error += np.sum(dft_calculation.stress**2.0)
return math.sqrt(n_stress_sq_error / d_stress_sq_error)
def energy_objective_function(md_calculations, dft_calculations):
n_energy_sq_error = 0.0
d_energy_sq_error = 0.0
# cannot calculate energy error if only one set of calculations
if len(md_calculations) == 1:
return 0.0
for (md_calc_i, dft_calc_i), (md_calc_j, dft_calc_j) in itertools.combinations(zip(md_calculations, dft_calculations), 2):
n_energy_sq_error += ((md_calc_i.energy - md_calc_j.energy) - (dft_calc_i.energy - dft_calc_j.energy))**2.0
d_energy_sq_error += (dft_calc_i.energy - dft_calc_j.energy)**2.0
return math.sqrt(n_energy_sq_error / d_energy_sq_error)
# material properties
def lattice_constant_objective_function(lattice, measured_lattice_constants):
# not going to include angles in calculation (different units need to reconcile)
lengths, angles = lattice.lengths_and_angles
n_lc_sq_error = 0.0
d_lc_sq_error = 0.0
for m_lengths, m_angles in measured_lattice_constants:
n_lc_sq_error += np.sum((lengths - m_lengths)**2.0)
d_lc_sq_error += np.sum(m_lengths**2.0)
return math.sqrt(n_lc_sq_error / d_lc_sq_error)
def elastic_constants_objective_function(elastic_constants, measured_elastic_constants):
elastic_constants_voigt = elastic_constants.voigt
n_ec_sq_error = 0.0
d_ec_sq_error = 0.0
for m_elastic_constants in measured_elastic_constants:
n_ec_sq_error = np.sum((elastic_constants_voigt - m_elastic_constants)**2.0)
d_ec_sq_error = np.sum(m_elastic_constants**2.0)
return math.sqrt(n_ec_sq_error / d_ec_sq_error)
def bulk_modulus_objective_function(elastic_constants, measured_bulk_modulus):
bulk_modulus = elastic_constants.k_vrh # Voigt-Reuss-Hill average bulk modulus
n_bm_sq_error = 0.0
d_bm_sq_error = 0.0
for m_bulk_modulus in measured_bulk_modulus:
n_bm_sq_error += (m_bulk_modulus - bulk_modulus)**2.0
d_bm_sq_error += m_bulk_modulus**2.0
return math.sqrt(n_bm_sq_error / d_bm_sq_error)
def shear_modulus_objective_function(elastic_constants, measured_shear_modulus):
shear_modulus = elastic_constants.g_vrh # Voigt-Reuss-Hill average shear modulus
n_sm_sq_error = 0.0
d_sm_sq_error = 0.0
for m_shear_modulus in measured_shear_modulus:
n_sm_sq_error += (m_shear_modulus - shear_modulus)**2.0
d_sm_sq_error += m_shear_modulus**2.0
return math.sqrt(n_sm_sq_error / d_sm_sq_error)