├── .gitignore ├── images ├── cosine-annealing.png ├── implementation1-learning-curve.png ├── implementation1-precision-recall.png ├── implementation2-precision-recall.png ├── implementation4-precision-recall.png ├── implementation4b-precision-recall.png ├── implementation5-learning-curve.png ├── implementation5-precision-recall.png ├── implementation5b-precision-recall.png ├── implementation6-learning-curve.png ├── implementation7-learning-curve.png └── implementation7-precision-recall.png ├── readme.md ├── requirements.txt ├── run.sh └── src ├── __init__.py ├── config.py ├── ml ├── __init__.py ├── data_loader.py ├── data_loader_edges.py ├── data_loader_with_meta.py ├── mf.py ├── mf_bias.py ├── mf_bias_continuous.py ├── mf_continuous.py ├── skipgram.py ├── skipgram_with_meta.py ├── skipgram_with_meta_weighted.py ├── train_gensim_embedding.py ├── train_node2vec_embeddings.py ├── train_torch_embedding.py ├── train_torch_embedding_with_meta.py ├── train_torch_mf.py ├── train_torch_mf_bias.py ├── train_torch_mf_bias_continuous_edges.py ├── train_torch_mf_bias_edges.py ├── train_torch_mf_bias_edges_parallel.py ├── train_torch_mf_continuous_edges.py └── train_torch_mf_edges.py ├── parse ├── __init__.py └── parse_json.py ├── prep ├── __init__.py ├── prep_edges.py ├── prep_graph_samples.py ├── prep_meta.py ├── prep_node_relationship.py └── train_val_split.py ├── utils ├── __init__.py ├── io_utils.py └── logger.py └── viz ├── __init__.py ├── plot_results.py └── prep_results.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/.gitignore -------------------------------------------------------------------------------- /images/cosine-annealing.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/images/cosine-annealing.png -------------------------------------------------------------------------------- /images/implementation1-learning-curve.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/images/implementation1-learning-curve.png -------------------------------------------------------------------------------- /images/implementation1-precision-recall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/images/implementation1-precision-recall.png -------------------------------------------------------------------------------- /images/implementation2-precision-recall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/images/implementation2-precision-recall.png -------------------------------------------------------------------------------- /images/implementation4-precision-recall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/images/implementation4-precision-recall.png -------------------------------------------------------------------------------- /images/implementation4b-precision-recall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/images/implementation4b-precision-recall.png -------------------------------------------------------------------------------- /images/implementation5-learning-curve.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/images/implementation5-learning-curve.png -------------------------------------------------------------------------------- /images/implementation5-precision-recall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/images/implementation5-precision-recall.png -------------------------------------------------------------------------------- /images/implementation5b-precision-recall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/images/implementation5b-precision-recall.png -------------------------------------------------------------------------------- /images/implementation6-learning-curve.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/images/implementation6-learning-curve.png -------------------------------------------------------------------------------- /images/implementation7-learning-curve.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/images/implementation7-learning-curve.png -------------------------------------------------------------------------------- /images/implementation7-precision-recall.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/images/implementation7-precision-recall.png -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/readme.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/run.sh -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/config.py -------------------------------------------------------------------------------- /src/ml/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ml/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/data_loader.py -------------------------------------------------------------------------------- /src/ml/data_loader_edges.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/data_loader_edges.py -------------------------------------------------------------------------------- /src/ml/data_loader_with_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/data_loader_with_meta.py -------------------------------------------------------------------------------- /src/ml/mf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/mf.py -------------------------------------------------------------------------------- /src/ml/mf_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/mf_bias.py -------------------------------------------------------------------------------- /src/ml/mf_bias_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/mf_bias_continuous.py -------------------------------------------------------------------------------- /src/ml/mf_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/mf_continuous.py -------------------------------------------------------------------------------- /src/ml/skipgram.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/skipgram.py -------------------------------------------------------------------------------- /src/ml/skipgram_with_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/skipgram_with_meta.py -------------------------------------------------------------------------------- /src/ml/skipgram_with_meta_weighted.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/skipgram_with_meta_weighted.py -------------------------------------------------------------------------------- /src/ml/train_gensim_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/train_gensim_embedding.py -------------------------------------------------------------------------------- /src/ml/train_node2vec_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/train_node2vec_embeddings.py -------------------------------------------------------------------------------- /src/ml/train_torch_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/train_torch_embedding.py -------------------------------------------------------------------------------- /src/ml/train_torch_embedding_with_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/train_torch_embedding_with_meta.py -------------------------------------------------------------------------------- /src/ml/train_torch_mf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/train_torch_mf.py -------------------------------------------------------------------------------- /src/ml/train_torch_mf_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/train_torch_mf_bias.py -------------------------------------------------------------------------------- /src/ml/train_torch_mf_bias_continuous_edges.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/train_torch_mf_bias_continuous_edges.py -------------------------------------------------------------------------------- /src/ml/train_torch_mf_bias_edges.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/train_torch_mf_bias_edges.py -------------------------------------------------------------------------------- /src/ml/train_torch_mf_bias_edges_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/train_torch_mf_bias_edges_parallel.py -------------------------------------------------------------------------------- /src/ml/train_torch_mf_continuous_edges.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/train_torch_mf_continuous_edges.py -------------------------------------------------------------------------------- /src/ml/train_torch_mf_edges.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/ml/train_torch_mf_edges.py -------------------------------------------------------------------------------- /src/parse/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/parse/parse_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/parse/parse_json.py -------------------------------------------------------------------------------- /src/prep/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/prep/prep_edges.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/prep/prep_edges.py -------------------------------------------------------------------------------- /src/prep/prep_graph_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/prep/prep_graph_samples.py -------------------------------------------------------------------------------- /src/prep/prep_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/prep/prep_meta.py -------------------------------------------------------------------------------- /src/prep/prep_node_relationship.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/prep/prep_node_relationship.py -------------------------------------------------------------------------------- /src/prep/train_val_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/prep/train_val_split.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/io_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/utils/io_utils.py -------------------------------------------------------------------------------- /src/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/utils/logger.py -------------------------------------------------------------------------------- /src/viz/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/viz/plot_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/viz/plot_results.py -------------------------------------------------------------------------------- /src/viz/prep_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eugeneyan/recsys-nlp-graph/HEAD/src/viz/prep_results.py --------------------------------------------------------------------------------