src

• src package
  • Subpackages
    • src.rumboost package
      • Submodules
      • src.rumboost.datasets module
      • src.rumboost.metrics module
      • src.rumboost.nested_cross_nested module
      • src.rumboost.ordinal module
      • src.rumboost.post_process module
      • src.rumboost.rumboost module
      • src.rumboost.torch_functions module
      • src.rumboost.utility_plotting module
      • src.rumboost.utility_smoothing module
      • src.rumboost.utils module
      • Module contents
    • src.statistical_models package
      • Submodules
      • src.statistical_models.mixed_effects module
      • Module contents
    • src.tastenet package
      • Submodules
      • src.tastenet.data_utils module
      • src.tastenet.models module
      • Module contents
  • Submodules
  • src.constants module
  • src.helper module
    • set_all_seeds()
  • src.hyperparameter_search module
    • main()
    • objective()
  • src.main module
    • main()
  • src.models_wrapper module
    • DNN
      • DNN.build_dataloader()
      • DNN.fit()
      • DNN.load_model()
      • DNN.predict()
      • DNN.save_model()
    • GBDT
      • GBDT.build_dataloader()
      • GBDT.fit()
      • GBDT.load_model()
      • GBDT.predict()
      • GBDT.save_model()
    • MixedEffect
      • MixedEffect.build_dataloader()
      • MixedEffect.fit()
      • MixedEffect.get_individual_parameters()
      • MixedEffect.load_model()
      • MixedEffect.predict()
      • MixedEffect.save_model()
    • RUMBoost
      • RUMBoost.build_dataloader()
      • RUMBoost.fit()
      • RUMBoost.load_model()
      • RUMBoost.predict()
      • RUMBoost.save_model()
    • TasteNet
      • TasteNet.build_dataloader()
      • TasteNet.fit()
      • TasteNet.load_model()
      • TasteNet.predict()
      • TasteNet.save_model()
  • src.parser module
    • parse_cmdline_args()
  • src.plotter module
    • plot_alt_spec_features()
    • plot_ind_spec_constant()
  • src.run_models module
  • src.split_save_dataset module
    • split_save_dataset()
  • src.synthetic_experiment module
    • add_simulated_choices()
    • compute_prob()
    • create_dataset()
    • create_functional_effects()
    • gather_functional_intercepts()
    • generate_labels()
    • generate_noise()
    • generate_x()
    • hyperparameter_search()
    • l1_distance()
    • plot_alt_spec_features()
    • plot_ind_spec_constant()
    • run_experiment()
    • utility_function()
  • src.train module
    • train()
  • src.utils module
    • add_hyperparameters()
    • build_lgb_dataset()
    • compute_metrics()
    • cross_entropy()
    • generate_general_params()
    • generate_ordinal_spec()
    • generate_rum_structure()
    • pkl_to_df()
    • split_dataset()
    • stratified_group_k_fold()
  • Module contents