LICENSE
README.md
pyproject.toml
setup.py
ptls/__init__.py
ptls/custom_layers.py
ptls/loss.py
ptls/make_datasets.py
ptls/make_datasets_spark.py
ptls/models.py
ptls/pl_fit_target.py
ptls/pl_inference.py
ptls/pl_inference_spark.py
ptls/pl_train_module.py
ptls/size_reduction.py
ptls/swa.py
ptls/tb_interface.py
ptls/util.py
ptls/data_load/__init__.py
ptls/data_load/fast_tensor_data_loader.py
ptls/data_load/feature_dict.py
ptls/data_load/filter_dataset.py
ptls/data_load/iterable_processing_dataset.py
ptls/data_load/list_splitter.py
ptls/data_load/padded_batch.py
ptls/data_load/partitioned_dataset.py
ptls/data_load/utils.py
ptls/data_load/augmentations/__init__.py
ptls/data_load/augmentations/all_time_shuffle.py
ptls/data_load/augmentations/build_augmentations.py
ptls/data_load/augmentations/drop_day.py
ptls/data_load/augmentations/dropout_trx.py
ptls/data_load/augmentations/random_slice.py
ptls/data_load/augmentations/seq_len_limit.py
ptls/data_load/augmentations/sequence_pair_augmentation.py
ptls/data_load/data_module/__init__.py
ptls/data_load/data_module/cls_data_module.py
ptls/data_load/data_module/coles_data_module.py
ptls/data_load/data_module/cpc_data_module.py
ptls/data_load/data_module/cpc_v2_data_module.py
ptls/data_load/data_module/emb_data_module.py
ptls/data_load/data_module/emb_valid_data_module.py
ptls/data_load/data_module/map_augmentation_dataset.py
ptls/data_load/data_module/nsp_data_module.py
ptls/data_load/data_module/rtd_data_module.py
ptls/data_load/data_module/seq_to_target_data_module.py
ptls/data_load/data_module/sop_data_module.py
ptls/data_load/datasets/__init__.py
ptls/data_load/datasets/augmentation_dataset.py
ptls/data_load/datasets/dataloaders.py
ptls/data_load/datasets/memory_dataset.py
ptls/data_load/datasets/parquet_dataset.py
ptls/data_load/datasets/parquet_file_scan.py
ptls/data_load/datasets/persist_dataset.py
ptls/data_load/iterable_processing/__init__.py
ptls/data_load/iterable_processing/category_size_clip.py
ptls/data_load/iterable_processing/feature_filter.py
ptls/data_load/iterable_processing/feature_rename.py
ptls/data_load/iterable_processing/feature_type_cast.py
ptls/data_load/iterable_processing/filter_non_array.py
ptls/data_load/iterable_processing/id_filter.py
ptls/data_load/iterable_processing/id_filter_df.py
ptls/data_load/iterable_processing/iterable_seq_len_limit.py
ptls/data_load/iterable_processing/iterable_shuffle.py
ptls/data_load/iterable_processing/seq_len_filter.py
ptls/data_load/iterable_processing/take_first_trx.py
ptls/data_load/iterable_processing/target_empty_filter.py
ptls/data_load/iterable_processing/target_extractor.py
ptls/data_load/iterable_processing/target_join.py
ptls/data_load/iterable_processing/target_move.py
ptls/data_load/iterable_processing/to_torch_tensor.py
ptls/frames/__init__.py
ptls/frames/abs_module.py
ptls/frames/inference_module.py
ptls/frames/ptls_data_module.py
ptls/frames/bert/__init__.py
ptls/frames/bert/datasets/__init__.py
ptls/frames/bert/datasets/mlm_dataset.py
ptls/frames/bert/datasets/mlm_indexed_dataset.py
ptls/frames/bert/datasets/nsp_dataset.py
ptls/frames/bert/datasets/rtd_dataset.py
ptls/frames/bert/datasets/sop_dataset.py
ptls/frames/bert/losses/__init__.py
ptls/frames/bert/losses/query_soft_max.py
ptls/frames/bert/modules/__init__.py
ptls/frames/bert/modules/mlm_module.py
ptls/frames/bert/modules/rtd_module.py
ptls/frames/bert/modules/sop_nsp_module.py
ptls/frames/coles/__init__.py
ptls/frames/coles/coles_dataset.py
ptls/frames/coles/coles_module.py
ptls/frames/coles/coles_supervised_dataset.py
ptls/frames/coles/coles_supervised_module.py
ptls/frames/coles/metric.py
ptls/frames/coles/split_strategy.py
ptls/frames/coles/losses/__init__.py
ptls/frames/coles/losses/barlow_twins_loss.py
ptls/frames/coles/losses/binomial_deviance_loss.py
ptls/frames/coles/losses/centroid_loss.py
ptls/frames/coles/losses/complex_loss.py
ptls/frames/coles/losses/contrastive_loss.py
ptls/frames/coles/losses/histogram_loss.py
ptls/frames/coles/losses/margin_loss.py
ptls/frames/coles/losses/triplet_loss.py
ptls/frames/coles/losses/vicreg_loss.py
ptls/frames/coles/sampling_strategies/__init__.py
ptls/frames/coles/sampling_strategies/all_positive_pair_selector.py
ptls/frames/coles/sampling_strategies/all_triplets_selector.py
ptls/frames/coles/sampling_strategies/distance_weighted_pair_selector.py
ptls/frames/coles/sampling_strategies/hard_negative_pair_selector.py
ptls/frames/coles/sampling_strategies/hard_triplet_selector.py
ptls/frames/coles/sampling_strategies/pair_selector.py
ptls/frames/coles/sampling_strategies/random_negative_triplet_selector.py
ptls/frames/coles/sampling_strategies/semi_hard_triplet_selector.py
ptls/frames/coles/sampling_strategies/triplet_selector.py
ptls/frames/cpc/__init__.py
ptls/frames/cpc/losses/__init__.py
ptls/frames/cpc/losses/cpc_loss.py
ptls/frames/cpc/metrics/__init__.py
ptls/frames/cpc/metrics/cpc_accuracy.py
ptls/frames/cpc/modules/__init__.py
ptls/frames/cpc/modules/cpc_module.py
ptls/frames/cpc/modules/cpc_v2_module.py
ptls/frames/supervised/__init__.py
ptls/frames/supervised/metrics.py
ptls/frames/supervised/seq_to_target.py
ptls/frames/supervised/seq_to_target_dataset.py
ptls/frames/tabformer/__init__.py
ptls/frames/tabformer/tabformer_dataset.py
ptls/frames/tabformer/tabformer_module.py
ptls/metric_learn/__init__.py
ptls/metric_learn/ml_models.py
ptls/metric_learn/read_processing.py
ptls/metric_learn/dataset/__init__.py
ptls/metric_learn/dataset/complex_target_dataset.py
ptls/metric_learn/dataset/splitting_dataset.py
ptls/metric_learn/dataset/target_enumerator_dataset.py
ptls/nn/__init__.py
ptls/nn/binarization.py
ptls/nn/head.py
ptls/nn/normalization.py
ptls/nn/pb.py
ptls/nn/pb_feature_extract.py
ptls/nn/seq_step.py
ptls/nn/seq_encoder/__init__.py
ptls/nn/seq_encoder/abs_seq_encoder.py
ptls/nn/seq_encoder/agg_feature_seq_encoder.py
ptls/nn/seq_encoder/containers.py
ptls/nn/seq_encoder/longformer_encoder.py
ptls/nn/seq_encoder/rnn_encoder.py
ptls/nn/seq_encoder/rnn_seq_encoder_distribution_target.py
ptls/nn/seq_encoder/statistics_encoder.py
ptls/nn/seq_encoder/transformer_encoder.py
ptls/nn/seq_encoder/utils.py
ptls/nn/trx_encoder/__init__.py
ptls/nn/trx_encoder/batch_norm.py
ptls/nn/trx_encoder/float_positional_encoding.py
ptls/nn/trx_encoder/noisy_embedding.py
ptls/nn/trx_encoder/scalers.py
ptls/nn/trx_encoder/tabformer_feature_encoder.py
ptls/nn/trx_encoder/trx_encoder.py
ptls/nn/trx_encoder/trx_encoder_base.py
ptls/nn/trx_encoder/trx_encoder_ohe.py
ptls/nn/trx_encoder/trx_mean_encoder.py
ptls/preprocessing/__init__.py
ptls/preprocessing/pandas_preprocessor.py
ptls/preprocessing/pyspark_preprocessor.py
ptls/preprocessing/util.py
ptls/preprocessing/base/__init__.py
ptls/preprocessing/base/col_category_transformer.py
ptls/preprocessing/base/col_transformer.py
ptls/preprocessing/base/data_preprocessor.py
ptls/preprocessing/pandas/__init__.py
ptls/preprocessing/pandas/category_identity_encoder.py
ptls/preprocessing/pandas/col_identity_transformer.py
ptls/preprocessing/pandas/col_transformer.py
ptls/preprocessing/pandas/event_time.py
ptls/preprocessing/pandas/frequency_encoder.py
ptls/preprocessing/pandas/user_group_transformer.py
ptls/preprocessing/pyspark/__init__.py
ptls/preprocessing/pyspark/category_identity_encoder.py
ptls/preprocessing/pyspark/col_identity_transformer.py
ptls/preprocessing/pyspark/col_transformer.py
ptls/preprocessing/pyspark/event_time.py
ptls/preprocessing/pyspark/frequency_encoder.py
ptls/preprocessing/pyspark/user_group_transformer.py
ptls_tests/__init__.py
ptls_tests/test_custom_layers.py
ptls_tests/test_distribution_target_loss.py
ptls_tests/test_list_subset.py
ptls_tests/test_loss.py
ptls_tests/test_pl_api.py
ptls_tests/test_ranking_loss.py
ptls_tests/test_size_reduction.py
ptls_tests/metric_learning_tests/__init__.py
ptls_tests/metric_learning_tests/test_collate.py
ptls_tests/test_data_load/__init__.py
ptls_tests/test_data_load/test__init__.py
ptls_tests/test_data_load/test_feature_dict.py
ptls_tests/test_data_load/test_list_splitter.py
ptls_tests/test_data_load/test_padded_batch.py
ptls_tests/test_data_load/test_utils.py
ptls_tests/test_data_load/test_augmentations/__init__.py
ptls_tests/test_data_load/test_augmentations/test_all_time_shuffle.py
ptls_tests/test_data_load/test_augmentations/test_dropout_trx.py
ptls_tests/test_data_load/test_augmentations/test_random_slice.py
ptls_tests/test_data_load/test_augmentations/test_seq_len_limit.py
ptls_tests/test_data_load/test_iterable_processing/__init__.py
ptls_tests/test_data_load/test_iterable_processing/test_category_size_clip.py
ptls_tests/test_data_load/test_iterable_processing/test_feature_rename.py
ptls_tests/test_data_load/test_iterable_processing/test_id_filter.py
ptls_tests/test_data_load/test_iterable_processing/test_id_filter_df.py
ptls_tests/test_data_load/test_iterable_processing/test_iterable_shuffle.py
ptls_tests/test_data_load/test_iterable_processing/test_seq_len_filter.py
ptls_tests/test_data_load/test_iterable_processing/test_target_empty_filter.py
ptls_tests/test_data_load/test_iterable_processing/test_target_extractor.py
ptls_tests/test_data_load/test_iterable_processing/test_target_join.py
ptls_tests/test_data_load/test_iterable_processing/test_target_move.py
ptls_tests/test_data_load/test_iterable_processing/test_to_torch.py
ptls_tests/test_frames/__init__.py
ptls_tests/test_frames/test_common_usage.py
ptls_tests/test_frames/test_cpc.py
ptls_tests/test_frames/test_inference_module.py
ptls_tests/test_frames/test_bert/__init__.py
ptls_tests/test_frames/test_bert/test_mlm.py
ptls_tests/test_frames/test_bert/test_datasets/__init__.py
ptls_tests/test_frames/test_bert/test_datasets/test_mlm_dataset.py
ptls_tests/test_frames/test_coles/__init__.py
ptls_tests/test_frames/test_coles/test_coles_module.py
ptls_tests/test_frames/test_coles/test_losses.py
ptls_tests/test_frames/test_coles/test_metrics.py
ptls_tests/test_frames/test_coles/test_sampling_strategies/__init__.py
ptls_tests/test_frames/test_coles/test_sampling_strategies/test_pair_selectors.py
ptls_tests/test_frames/test_coles/test_sampling_strategies/test_triplet_selectors.py
ptls_tests/test_frames/test_supervised/__init__.py
ptls_tests/test_frames/test_supervised/test_metrics.py
ptls_tests/test_frames/test_supervised/test_seq_to_target.py
ptls_tests/test_frames/test_supervised/test_seq_to_target_dataset.py
ptls_tests/test_nn/__init__.py
ptls_tests/test_nn/test_head.py
ptls_tests/test_nn/test_normalization.py
ptls_tests/test_nn/test_pb.py
ptls_tests/test_nn/test_seq_step.py
ptls_tests/test_nn/test_seq_encoder/__init__.py
ptls_tests/test_nn/test_seq_encoder/test_agg_feature_seq_encoder.py
ptls_tests/test_nn/test_seq_encoder/test_containers.py
ptls_tests/test_nn/test_seq_encoder/test_longformer_encoder.py
ptls_tests/test_nn/test_seq_encoder/test_rnn_encoder.py
ptls_tests/test_nn/test_seq_encoder/test_rnn_seq_distribution_targets_encoder.py
ptls_tests/test_nn/test_seq_encoder/test_statistics_encoder.py
ptls_tests/test_nn/test_seq_encoder/test_transformer_encoder.py
ptls_tests/test_nn/test_seq_encoder/test_utils.py
ptls_tests/test_nn/test_trx_encoder/__init__.py
ptls_tests/test_nn/test_trx_encoder/test_batch_norm.py
ptls_tests/test_nn/test_trx_encoder/test_noisy_embedding.py
ptls_tests/test_nn/test_trx_encoder/test_trx_encoder.py
ptls_tests/test_nn/test_trx_encoder/test_trx_encoder_base.py
ptls_tests/test_nn/test_trx_encoder/test_trx_mean_encoder.py
ptls_tests/test_nn/test_trx_encoder/test_trx_ohe_encoder.py
ptls_tests/utils/__init__.py
ptls_tests/utils/data_generation.py
pytorch_lifestream.egg-info/PKG-INFO
pytorch_lifestream.egg-info/SOURCES.txt
pytorch_lifestream.egg-info/dependency_links.txt
pytorch_lifestream.egg-info/requires.txt
pytorch_lifestream.egg-info/top_level.txt