feature_matrix_from_entityset.py

def feature_matrix_from_entityset(es_dict, feature_defs):
    """Run deep feature synthesis from an entityset and feature definitions. 
       Saves feature matrix based on partition number.""" 
    
    # Extract the entityset
    es = es_dict['es']
    
    # Calculate the feature matrix and save
    feature_matrix = ft.calculate_feature_matrix(feature_definitions, 
                                                 entityset=es, 
                                                 n_jobs = 1, 
                                                 verbose = 0,
                                                 chunk_size = es['app'].df.shape[0])
    
    feature_matrix.to_csv('../input/fm/p%d_fm.csv' % es_dict['num'], index = True)