Thesis-SampleAnalysis/modules/features.py

from . import config as config

import numpy as np

def compute_feature(feature_choice, samples):

    data = []

    if feature_choice == 'samples':
        data = samples

    if feature_choice == 'variances':

        incr_samples = []
        
        # evolution of variance
        for sample in samples:
            incr_samples.append(sample)
            data.append(np.var(incr_samples))

    return data