def get_feature_vector(hypers, frames):
with ostream_redirect():
soap = SphericalCovariants(**hypers)
soap_vectors = soap.transform(frames)
print('Feature vector size: %.3fMB' %
(soap.get_num_coefficients()*8.0/1.0e6))
feature_vector = soap_vectors.get_features(soap)
return feature_vector
def get_soap_vectors(hypers, frames):
with ostream_redirect():
sph_expn = SphericalExpansion(**hypers)
expansions = sph_expn.transform(frames)
soap_vectors = expansions.get_features(sph_expn)
return soap_vectors