mattsgithub

import numpy as np
import pandas as pd

def get_disk_holes(r, r_hole, n_hole, allow_center=True):
    """
    Returns disks that are can be used to
    cut data from a disk of radius `r`
    """
    
    df = pd.DataFrame(columns=['r_hole', 'r', 'theta', 'x', 'y'])

mattsgithub

mattsgithub

def get_pandas_df_for_counterfactual_prediction(df,
                                                df_treatments,
                                                id_column='customer_id',
                                                suffix='_assigned'):
    """Returns a pandas dataframe prepared for counterfactual predictions
    
    Args
        df (pd.DataFrame):
            The dataframe from which to prepare counterfactuals for
        

mattsgithub

class SampleForwardChainCV(object):
    def __init__(self,
                 dates,
                 obs_count,
                 min_start_date=None,
                 max_end_date=None,
                 n_min_train_obs=20,
                 n_min_validate_obs=20,
                 n_min_test_obs=20):

mattsgithub

mattsgithub

mattsgithub