synicix_ml_pipeline.dataset_classes package¶

Submodules¶

synicix_ml_pipeline.dataset_classes.NeuroDataDataset module¶

class synicix_ml_pipeline.dataset_classes.NeuroDataDataset.NeuroDataDataset(tuple_dict, dataset_cache_dir, tier, mode)[source]¶

Bases: torch.utils.data.dataset.Dataset

get_additional_model_params()[source]¶

Return additional_model_params

Parameters:: None
Returns:: dict : additional_model_params to be passed on to the model at creation time

synicix_ml_pipeline.dataset_classes.StockDataset module¶

class synicix_ml_pipeline.dataset_classes.StockDataset.StockDataset(tuple_dict, dataset_cache_dir, tier, train_percentage, validation_percentage, inputs_columns=['Price', 'Open', 'High', 'Low', 'Vol.'], target_columns=['Change %'], input_time_window_len=30, target_time_window_len=1)[source]¶