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