Skip to content

Commit cd8c858

Browse files
Added feature rescaling to vectorized LDOS reconstruction
1 parent ba4b814 commit cd8c858

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

mala/datahandling/ldos_feature_extractor.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -269,6 +269,12 @@ def reconstruct_ldos(self, ldos_features, original_dimension=300):
269269
)
270270
elif len(ldos_shape) != 2:
271271
raise Exception("Cannot work with provided LDOS shape.")
272+
273+
if self.use_feature_scaling:
274+
for f in range(5 + 3 * self.number_of_gaussians):
275+
ldos_features[:, f] = self.__internal_feature_rescaling(
276+
ldos_features[:, f], f
277+
)
272278
full_ldos = np.zeros((ldos_features.shape[0], original_dimension))
273279

274280
# Initializing the arrays for the left and right side of linear fit as

0 commit comments

Comments
 (0)