Path	Lines of Code
g3doc/_book.yaml	31
g3doc/_index.yaml	61
neural_structured_learning/__init__.py	7
neural_structured_learning/configs/__init__.py	42
neural_structured_learning/configs/configs.py	151
neural_structured_learning/estimator/__init__.py	4
neural_structured_learning/estimator/adversarial_regularization.py	59
neural_structured_learning/estimator/graph_regularization.py	73
neural_structured_learning/experimental/__init__.py	1
neural_structured_learning/experimental/gnn.py	195
neural_structured_learning/keras/__init__.py	8
neural_structured_learning/keras/adversarial_regularization.py	363
neural_structured_learning/keras/graph_regularization.py	44
neural_structured_learning/keras/layers/__init__.py	6
neural_structured_learning/keras/layers/neighbor_features.py	76
neural_structured_learning/keras/layers/pairwise_distance.py	61
neural_structured_learning/lib/__init__.py	34
neural_structured_learning/lib/abstract_gen_neighbor.py	8
neural_structured_learning/lib/adversarial_neighbor.py	178
neural_structured_learning/lib/distances.py	141
neural_structured_learning/lib/regularizer.py	43
neural_structured_learning/lib/utils.py	253
neural_structured_learning/tools/__init__.py	7
neural_structured_learning/tools/build_docs.py	35
neural_structured_learning/tools/build_graph.py	180
neural_structured_learning/tools/graph_utils.py	50
neural_structured_learning/tools/pack_nbrs.py	119
neural_structured_learning/version.py	7
research/a2n/__init__.py	1
research/a2n/dataset.py	352
research/a2n/encoders.py	412
research/a2n/generate_random_graph.py	26
research/a2n/graph.py	265
research/a2n/losses.py	9
research/a2n/metrics.py	30
research/a2n/models.py	316
research/a2n/train.py	533
research/a2n/utils.py	55
research/carls/__init__.py	26
research/carls/base/async_node_hash_map.h	256
research/carls/base/embedding_helper.cc	103
research/carls/base/embedding_helper.h	29
research/carls/base/file_helper.cc	101
research/carls/base/file_helper.h	23
research/carls/base/input_context_helper.cc	652
research/carls/base/input_context_helper.h	103
research/carls/base/proto_factory.h	104
research/carls/base/proto_helper.cc	115
research/carls/base/proto_helper.h	36
research/carls/base/status_helper.cc	163
research/carls/base/status_helper.h	105
research/carls/base/thread_bundle.cc	57
research/carls/base/thread_bundle.h	29
research/carls/base/top_n.h	155
research/carls/bazel/build_rules.bzl	430
research/carls/bazel/repo.bzl	287
research/carls/candidate_sampling/brute_force_topk_sampler.cc	117
research/carls/candidate_sampling/candidate_sampler.cc	29
research/carls/candidate_sampling/candidate_sampler.h	37
research/carls/candidate_sampling/candidate_sampler_config.proto	73
research/carls/candidate_sampling/candidate_sampler_config_builder.py	36
research/carls/candidate_sampling/negative_sampler.cc	242
research/carls/candidate_sampling_ops.py	93
research/carls/constants.h	6
research/carls/context.py	25
research/carls/dynamic_embedding_config.proto	16
research/carls/dynamic_embedding_manager.cc	424
research/carls/dynamic_embedding_manager.h	54
research/carls/dynamic_embedding_neighbor_cache.py	30
research/carls/dynamic_embedding_ops.py	69
research/carls/dynamic_memory_ops.py	36
research/carls/dynamic_normalization.py	160
research/carls/embedding.proto	20
research/carls/gradient_descent/gradient_descent_config.proto	20
research/carls/gradient_descent/gradient_descent_optimizer.cc	110
research/carls/gradient_descent/gradient_descent_optimizer.h	35
research/carls/graph_regularization.py	49
research/carls/input_context.proto	49
research/carls/io_ops.py	29
research/carls/kbs_server_helper.cc	116
research/carls/kbs_server_helper.h	30
research/carls/kbs_server_helper_pybind.cc	22
research/carls/kernels/__init__.py	8
research/carls/kernels/dynamic_embedding_manager_resource.cc	82
research/carls/kernels/dynamic_embedding_manager_resource.h	20
research/carls/kernels/dynamic_embedding_ops.cc	236
research/carls/kernels/dynamic_memory_ops.cc	100
research/carls/kernels/io_ops.cc	107
research/carls/kernels/sampled_logits_ops.cc	162
research/carls/kernels/topk_ops.cc	76
research/carls/knowledge_bank/in_proto_knowledge_bank.cc	126
research/carls/knowledge_bank/initializer.proto	28
research/carls/knowledge_bank/initializer_helper.cc	108
research/carls/knowledge_bank/initializer_helper.h	18
research/carls/knowledge_bank/knowledge_bank.cc	118
research/carls/knowledge_bank/knowledge_bank.h	53
research/carls/knowledge_bank/knowledge_bank_config.proto	42
research/carls/knowledge_bank/leveldb_knowledge_bank.cc	228
research/carls/knowledge_bank_grpc_service.cc	363
research/carls/knowledge_bank_grpc_service.h	57
research/carls/knowledge_bank_service.proto	125
research/carls/memory_store/distance_helper.cc	37
research/carls/memory_store/distance_helper.h	12
research/carls/memory_store/gaussian_memory.cc	328
research/carls/memory_store/gaussian_memory_config.proto	47
research/carls/memory_store/memory_distance_config.proto	17
research/carls/memory_store/memory_store.cc	53
research/carls/memory_store/memory_store.h	48
research/carls/memory_store/memory_store_config.proto	19
research/carls/models/caml/sparse_features.py	247
research/carls/models/caml/sparse_features_config.proto	72
research/carls/neighbor_cache_client.py	15
research/carls/util/array_ops.py	7
research/gam/gam/__init__.py	1
research/gam/gam/data/__init__.py	1
research/gam/gam/data/dataset.py	644
research/gam/gam/data/loaders.py	160
research/gam/gam/data/preprocessing.py	55
research/gam/gam/data/robustness.py	64
research/gam/gam/experiments/__init__.py	1
research/gam/gam/experiments/helper.py	112
research/gam/gam/experiments/run_train_gam.py	395
research/gam/gam/experiments/run_train_gam_graph.py	371
research/gam/gam/models/__init__.py	1
research/gam/gam/models/cnn.py	153
research/gam/gam/models/gcn.py	173
research/gam/gam/models/mlp.py	104
research/gam/gam/models/models_base.py	73
research/gam/gam/models/wide_resnet.py	179
research/gam/gam/trainer/__init__.py	1
research/gam/gam/trainer/adversarial_dense.py	59
research/gam/gam/trainer/adversarial_sparse.py	82
research/gam/gam/trainer/trainer_agreement.py	656
research/gam/gam/trainer/trainer_base.py	95
research/gam/gam/trainer/trainer_classification.py	602
research/gam/gam/trainer/trainer_classification_gcn.py	579
research/gam/gam/trainer/trainer_cotrain.py	414
research/gnn-survey/layers.py	167
research/gnn-survey/models.py	168
research/gnn-survey/train.py	84
research/gnn-survey/utils.py	122
research/kg_hyp_emb/__init__.py	1
research/kg_hyp_emb/config.py	44
research/kg_hyp_emb/datasets/__init__.py	1
research/kg_hyp_emb/datasets/datasets.py	40
research/kg_hyp_emb/datasets/process.py	63
research/kg_hyp_emb/learning/__init__.py	1
research/kg_hyp_emb/learning/losses.py	35
research/kg_hyp_emb/learning/regularizers.py	33
research/kg_hyp_emb/learning/trainer.py	47
research/kg_hyp_emb/models/__init__.py	12
research/kg_hyp_emb/models/base.py	119
research/kg_hyp_emb/models/complex.py	55
research/kg_hyp_emb/models/euclidean.py	131
research/kg_hyp_emb/models/hyperbolic.py	115
research/kg_hyp_emb/train.py	133
research/kg_hyp_emb/utils/__init__.py	1
research/kg_hyp_emb/utils/euclidean.py	30
research/kg_hyp_emb/utils/hyperbolic.py	47
research/kg_hyp_emb/utils/train.py	32
research/multi_representation_adversary/multi_representation_adversary/__init__.py	1
research/multi_representation_adversary/multi_representation_adversary/attacks.py	178
research/multi_representation_adversary/multi_representation_adversary/cleverhans_adapter.py	9
research/multi_representation_adversary/multi_representation_adversary/data.py	29
research/multi_representation_adversary/multi_representation_adversary/evaluator.py	119
research/multi_representation_adversary/multi_representation_adversary/helper.py	47
research/multi_representation_adversary/multi_representation_adversary/main.py	38
research/multi_representation_adversary/multi_representation_adversary/resnet.py	86
research/multi_representation_adversary/multi_representation_adversary/selectors.py	79
research/multi_representation_adversary/multi_representation_adversary/trainer.py	93
research/multi_representation_adversary/multi_representation_adversary/transforms.py	44
research/neural_clustering/__init__.py	4
research/neural_clustering/data_generators/__init__.py	3
research/neural_clustering/data_generators/mog.py	55
research/neural_clustering/data_generators/partition.py	32
research/neural_clustering/models/__init__.py	8
research/neural_clustering/models/ncp_base.py	72
research/neural_clustering/models/ncp_models.py	27
research/neural_clustering/models/ncp_wrapper.py	44
research/neural_clustering/setup.py	11
research/neural_clustering/utils/__init__.py	3
research/neural_clustering/utils/data_utils.py	16
setup.cfg	10
setup.py	34