Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/JhuoW/ERIC
/ functions
Functions
241 in github.com/JhuoW/ERIC
⨍
Functions
241
◇
Types & classes
32
↓ 15 callers
Method
load
(self)
DataHelper/dataset_helper.py:32
↓ 15 callers
Method
log
(self , content = "")
utils/logger.py:47
↓ 8 callers
Function
calculate_prec_at_k
Calculating precision at k.
utils/utils.py:210
↓ 8 callers
Function
calculate_ranking_correlation
Calculating specific ranking correlation for predicted values. :param rank_corr_function: Ranking correlation function. :param prediction
utils/utils.py:181
↓ 8 callers
Function
get_positive_expectation
Computes the positive part of a divergence / difference. Args: p_samples: Positive samples. 一个矩阵 [n_nodes, n_graphs] 每个节点和它所在的图的相似度, 其
utils/gan_losses.py:16
↓ 8 callers
Function
save_fig
(plt, dir, fn, print_path=True)
utils/utils.py:42
↓ 6 callers
Method
add_line
(self , num = -1 , char = "-")
utils/logger.py:58
↓ 6 callers
Method
close
(self)
utils/logger.py:43
↓ 6 callers
Method
dist_metric
(self)
DataHelper/simgnn_utils.py:520
↓ 6 callers
Method
transform_batch
Getting ged for graph pair and grouping with data into dictionary. :param data: Graph pair. :return new_data: Dictionary with
DataHelper/DatasetLocal.py:164
↓ 5 callers
Method
__init__
(self, input_dim)
model/layers.py:211
↓ 5 callers
Function
get_config
(config_path="config.yml")
utils/utils.py:16
↓ 5 callers
Function
save_best_val_model_metric
(config, dataset_name, model, PATH_MODEL, metric)
utils/utils.py:255
↓ 4 callers
Function
_list_safe_get
(l, index, default)
utils/vis.py:187
↓ 4 callers
Function
get_true_result
(all_graphs, testing_graphs, trainval_graphs, sim_or_dist = 'dist')
exp/exp.py:58
↓ 4 callers
Function
load_data
(args, custom = False)
utils/utils.py:22
↓ 4 callers
Method
preprocess
(self, gs, max_num_nodes)
DataHelper/CustomDataset.py:121
↓ 3 callers
Method
_load_result_mat
(self, metric, model, m, n)
DataHelper/simgnn_utils.py:552
↓ 3 callers
Method
_select_dist_mat
(self, norm)
DataHelper/simgnn_utils.py:605
↓ 3 callers
Function
_sorted_dict
(d)
utils/vis.py:133
↓ 3 callers
Method
dist_sim_mat
Each result object stores either a distance matrix or a similarity matrix. It cannot store both. :param norm: :re
DataHelper/simgnn_utils.py:377
↓ 3 callers
Method
get_dataset_folder_name
(self)
DataHelper/simgnn_utils.py:76
↓ 3 callers
Function
plot_embeddings_for_points
(points, what_gs,extra_dir)
exp/exp.py:291
↓ 2 callers
Method
__init__
(self, save_folder)
DataHelper/simgnn_utils.py:38
↓ 2 callers
Function
_calculate_prec_at_k
(k, target)
utils/utils.py:201
↓ 2 callers
Method
_check_graphs_num
(graphs, label)
DataHelper/simgnn_utils.py:165
↓ 2 callers
Method
_node_feat_dic
(self, g)
DataHelper/simgnn_utils.py:204
↓ 2 callers
Method
_shuffle
(self)
DataHelper/simgnn_utils.py:33
↓ 2 callers
Method
calculate_distance_btw_pairs
(self, g1, g2, scale)
DataHelper/simgnn_utils.py:345
↓ 2 callers
Method
collect_embeddings
(self, all_graphs)
model/GSC.py:156
↓ 2 callers
Method
collect_graph_embeddings
(self, all_graphs)
model/GSC.py:169
↓ 2 callers
Method
convolutional_pass
Making convolutional pass. :param edge_index: Edge indices. :param features: Feature matrix. :return features: Abstra
model/SImGNN.py:121
↓ 2 callers
Method
convolutional_pass_level
(self, enc, edge_index, x)
model/GSC.py:144
↓ 2 callers
Method
convolutional_pass_level
(self, enc, edge_index, x)
model/Base/GSC_custom.py:118
↓ 2 callers
Function
create_dir_if_not_exists
(dir)
utils/utils.py:36
↓ 2 callers
Method
deepsets_outer
(self, batch, feat, filter_idx, size = None)
model/GSC.py:150
↓ 2 callers
Method
deepsets_outer
(self, batch, feat, filter_idx, size = None)
model/Base/GSC_custom.py:124
↓ 2 callers
Method
diffpool
Making differentiable pooling. :param abstract_features: Node feature matrix. :param edge_index: Edge indices :param
model/SImGNN.py:137
↓ 2 callers
Function
draw_extra
(i, ax, info_dict, text)
utils/vis.py:179
↓ 2 callers
Function
draw_graph_small
(g, info_dict)
utils/vis.py:99
↓ 2 callers
Function
get_color_map
(gs, trainval_graphs, use_color = True)
exp/exp.py:588
↓ 2 callers
Method
get_dataset_folder_name
(self)
DataHelper/simgnn_utils.py:102
↓ 2 callers
Method
get_max_number_of_nodes
(self)
DataHelper/simgnn_utils.py:49
↓ 2 callers
Method
get_save_folder
(self)
DataHelper/simgnn_utils.py:43
↓ 2 callers
Function
get_text_label_for_ranking
(ds_metric, qid, gid, norm, is_query, dataset, gids_groundtruth, plot_gids, nor
exp/exp.py:410
↓ 2 callers
Function
iterate_get_graphs
(directory)
DataHelper/simgnn_utils.py:238
↓ 2 callers
Method
load_data_set
(self, data, train_or_test)
DataHelper/simgnn_utils.py:142
↓ 2 callers
Function
normalized_exp_kernel_ged
(d, g1, g2, scale)
DataHelper/simgnn_utils.py:252
↓ 2 callers
Function
print_evaluation
Printing the error rates.
main.py:170
↓ 2 callers
Function
proc_filepath
(filepath)
DataHelper/simgnn_utils.py:257
↓ 2 callers
Function
raise_measure_error
(measure)
utils/gan_losses.py:9
↓ 2 callers
Function
set_node_attr
(g, types)
utils/vis.py:16
↓ 2 callers
Method
sort_id_mat
:param norm: :return: a m by n matrix representing the ranking result. rtn[i][j]: For query i, the id of the j-th most si
DataHelper/simgnn_utils.py:472
↓ 2 callers
Method
time
(self, qid, gid)
DataHelper/simgnn_utils.py:463
↓ 2 callers
Function
to_directed
(edge_index)
DataHelper/data_utils.py:6
↓ 2 callers
Function
top_k_ids
(query, qid, k, inclusive, rm=0)
DataHelper/simgnn_utils.py:275
↓ 1 callers
Function
Evaluation
(output, labels)
utils/utils.py:95
↓ 1 callers
Method
__init__
(self, nfeat, nhid, nclass, num_layers = 2 ,use_bn=True)
model/MLP.py:8
↓ 1 callers
Method
__init__
(self, config, n_feat)
model/GSC.py:17
↓ 1 callers
Method
__init__
(self, config, n_feat)
model/Base/GSC_custom.py:11
↓ 1 callers
Function
_calc_subplot_size_small
(area)
utils/vis.py:174
↓ 1 callers
Method
_choose_result_file
(self, files, m, n)
DataHelper/simgnn_utils.py:592
↓ 1 callers
Method
_dense_adj_normalization
(adj)
DataHelper/CustomDataset.py:151
↓ 1 callers
Function
_get_edge_width
(g, info_dict)
utils/vis.py:93
↓ 1 callers
Function
_get_line_width
(g)
utils/vis.py:88
↓ 1 callers
Function
_get_node_colors
(g, info_dict)
utils/vis.py:139
↓ 1 callers
Method
_get_node_feature_encoder
(self, graphs)
DataHelper/simgnn_utils.py:169
↓ 1 callers
Function
_get_node_size
(g, info_dict)
utils/vis.py:129
↓ 1 callers
Method
_graph_feature_adj
(self, gs, max_nodes)
DataHelper/CustomDataset.py:130
↓ 1 callers
Function
_info_dict_preprocess
(info_dict)
utils/vis.py:153
↓ 1 callers
Method
_load_merged_astar_from_other_three
(self, metric, m, n)
DataHelper/simgnn_utils.py:568
↓ 1 callers
Method
_load_test_triples
(self)
DataHelper/CustomDataset.py:171
↓ 1 callers
Method
_load_train_triples
(self, dist_calculator)
DataHelper/CustomDataset.py:159
↓ 1 callers
Method
_load_val_train_triples
(self, dist_calculator)
DataHelper/CustomDataset.py:190
↓ 1 callers
Function
_save_figs
(info_dict)
utils/vis.py:194
↓ 1 callers
Method
_select_sort_id_mat
(self, norm)
DataHelper/simgnn_utils.py:608
↓ 1 callers
Method
_train_val_split
(self, orig_train_data)
DataHelper/simgnn_utils.py:153
↓ 1 callers
Method
calculate_bottleneck_features
Deciding the shape of the bottleneck layer.
model/SImGNN.py:26
↓ 1 callers
Method
calculate_histogram
Calculate histogram from similarity matrix. :param abstract_features_1: Feature matrix for target graphs. :param abstract_fea
model/SImGNN.py:86
↓ 1 callers
Function
classification_mat
(thresh_pos, thresh_neg,ground_truth_nged, norm = True)
exp/exp.py:323
↓ 1 callers
Function
classify
(dist, thresh_pos, thresh_neg)
utils/utils.py:169
↓ 1 callers
Method
create_batches
(self, config)
DataHelper/DatasetLocal.py:136
↓ 1 callers
Method
dist_mat
(self, norm)
DataHelper/simgnn_utils.py:523
↓ 1 callers
Function
draw_emb_hist_heat_helper
(gnn_id, nel, cmap_color, dataset_name, test_graphs, trainval_graphs, ids, groun
exp/exp.py:107
↓ 1 callers
Method
encode
(self, g)
DataHelper/simgnn_utils.py:179
↓ 1 callers
Function
evaluate
(testing_graphs, training_graphs, model, loss_func, dataset: DatasetLocal, config)
test.py:12
↓ 1 callers
Function
evaluate
(testing_graphs, training_graphs, model, loss_func, dataset: DatasetLocal, config)
test_linux.py:12
↓ 1 callers
Function
evaluate
(testing_graphs, training_graphs, model, loss_func, dataset: DatasetLocal, config)
test_imdb.py:12
↓ 1 callers
Function
gen_pair
(g, kl=None, ku=2)
DataHelper/data_utils.py:12
↓ 1 callers
Function
gen_pairs
(graphs, kl=None, ku=2)
DataHelper/data_utils.py:50
↓ 1 callers
Function
get_classification_labels_from_dist_mat
(dist_mat, thresh_pos, thresh_neg)
utils/utils.py:144
↓ 1 callers
Function
get_ged_select_norm_str
(unnormed_matrix,normed_matrix, qid, gid, norm)
exp/exp.py:454
↓ 1 callers
Method
get_max_number_of_nodes
(self)
DataHelper/simgnn_utils.py:82
↓ 1 callers
Method
get_max_number_of_nodes
(self)
DataHelper/simgnn_utils.py:108
↓ 1 callers
Method
get_next_one
(self)
DataHelper/simgnn_utils.py:23
↓ 1 callers
Function
get_norm_str
(norm)
exp/exp.py:467
↓ 1 callers
Function
get_pred_result
(pred_mat, sim_or_dist = 'sim')
exp/exp.py:75
↓ 1 callers
Function
get_root_path
()
DataHelper/simgnn_utils.py:314
↓ 1 callers
Method
get_training_batch
(self)
DataHelper/CustomDataset.py:61
↓ 1 callers
Method
init_emb
(self)
model/GSC.py:282
next →
1–100 of 241, ranked by callers