Пример #1
0
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