Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
'user_pseudo_id': config.get('index_col'),
'event_name': config.get('event_col'),
'event_timestamp': config.get('event_time_col'),
}
if not os.path.exists(config['experiments_folder']):
os.mkdir(config['experiments_folder'])
@pd.api.extensions.register_dataframe_accessor("trajectory")
class RetentioneeringTrajectory(BaseTrajectory):
def __init__(self, pandas_obj):
super(RetentioneeringTrajectory, self).__init__(pandas_obj)
self.retention_config = config
@pd.api.extensions.register_dataframe_accessor("retention")
class RetentioneeringDataset(BaseDataset):
def __init__(self, pandas_obj):
super(RetentioneeringDataset, self).__init__(pandas_obj)
self.retention_config = config
def __init__(self, pandas_obj):
super(BaseDataset, self).__init__(pandas_obj)
self._embedding_types = ['tfidf', 'counts', 'frequency']
self._locals = None