build_data_generator

behavenet.data.utils.build_data_generator(hparams, sess_ids, export_csv=True)[source]

Helper function to build data generator from hparams dict.

Parameters:
  • hparams (dict) – needs to contain information specifying data inputs to model

  • sess_ids (list of dict) – each entry is a session dict with keys ‘lab’, ‘expt’, ‘animal’, ‘session’

  • export_csv (bool, optional) – export csv file containing session info (useful when fitting multi-sessions)

Returns:

data generator

Return type:

ConcatSessionsGenerator object