├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── data └── create_dataset.py ├── data_analysis_visualizations.ipynb ├── models ├── __init__.py ├── gain.py ├── gat.py ├── init_model.py └── pool.py ├── predicting-choice.ipynb ├── predicting-cities.ipynb ├── predicting-classes.ipynb ├── predicting-clusters.ipynb ├── predicting-roads.ipynb ├── prediction.py ├── requirements.txt ├── scripts ├── __init__.py ├── accident-hyperparam-search.py ├── class-hyperparam-search.py ├── clean_gdf.py ├── dataset-build.py ├── roads-feat-optim.py ├── roads-hyperparam-search.py ├── roads-prediction-dataset-gen-models.py └── roads-prediction-dataset-gen.py ├── train.py └── utils ├── __init__.py ├── constants.py ├── custom_link_split.py ├── custom_node_split.py ├── early_stopping.py ├── from_networkx.py ├── gdf_to_nx.py ├── load_data.py ├── load_geodata.py ├── metric.py ├── normalize_features.py ├── remove_false_nodes.py ├── seed.py ├── ssx_cities_dataset.py ├── utils.py ├── valid_edge.py ├── virtual_node.py └── visualization.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/create_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/data/create_dataset.py -------------------------------------------------------------------------------- /data_analysis_visualizations.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/data_analysis_visualizations.ipynb -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/gain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/models/gain.py -------------------------------------------------------------------------------- /models/gat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/models/gat.py -------------------------------------------------------------------------------- /models/init_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/models/init_model.py -------------------------------------------------------------------------------- /models/pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/models/pool.py -------------------------------------------------------------------------------- /predicting-choice.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/predicting-choice.ipynb -------------------------------------------------------------------------------- /predicting-cities.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/predicting-cities.ipynb -------------------------------------------------------------------------------- /predicting-classes.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/predicting-classes.ipynb -------------------------------------------------------------------------------- /predicting-clusters.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/predicting-clusters.ipynb -------------------------------------------------------------------------------- /predicting-roads.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/predicting-roads.ipynb -------------------------------------------------------------------------------- /prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/prediction.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/accident-hyperparam-search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/scripts/accident-hyperparam-search.py -------------------------------------------------------------------------------- /scripts/class-hyperparam-search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/scripts/class-hyperparam-search.py -------------------------------------------------------------------------------- /scripts/clean_gdf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/scripts/clean_gdf.py -------------------------------------------------------------------------------- /scripts/dataset-build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/scripts/dataset-build.py -------------------------------------------------------------------------------- /scripts/roads-feat-optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/scripts/roads-feat-optim.py -------------------------------------------------------------------------------- /scripts/roads-hyperparam-search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/scripts/roads-hyperparam-search.py -------------------------------------------------------------------------------- /scripts/roads-prediction-dataset-gen-models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/scripts/roads-prediction-dataset-gen-models.py -------------------------------------------------------------------------------- /scripts/roads-prediction-dataset-gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/scripts/roads-prediction-dataset-gen.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/constants.py -------------------------------------------------------------------------------- /utils/custom_link_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/custom_link_split.py -------------------------------------------------------------------------------- /utils/custom_node_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/custom_node_split.py -------------------------------------------------------------------------------- /utils/early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/early_stopping.py -------------------------------------------------------------------------------- /utils/from_networkx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/from_networkx.py -------------------------------------------------------------------------------- /utils/gdf_to_nx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/gdf_to_nx.py -------------------------------------------------------------------------------- /utils/load_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/load_data.py -------------------------------------------------------------------------------- /utils/load_geodata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/load_geodata.py -------------------------------------------------------------------------------- /utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/metric.py -------------------------------------------------------------------------------- /utils/normalize_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/normalize_features.py -------------------------------------------------------------------------------- /utils/remove_false_nodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/remove_false_nodes.py -------------------------------------------------------------------------------- /utils/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/seed.py -------------------------------------------------------------------------------- /utils/ssx_cities_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/ssx_cities_dataset.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/utils.py -------------------------------------------------------------------------------- /utils/valid_edge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/valid_edge.py -------------------------------------------------------------------------------- /utils/virtual_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/virtual_node.py -------------------------------------------------------------------------------- /utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wilsoncwc/predicting-choice/HEAD/utils/visualization.py --------------------------------------------------------------------------------