bupt-gamma / ciah Goto Github PK

It is an interesting work on clustering! So I have cloned and try to run it
However, an error occurs:

ImportError: cannot import name 'negative_sample' from 'Criterion'
Then I realized there is no exist the definition of negative_sample function in this repository
Maybe the code is forgotten to submit? I have expectations for the complete code version.

Hope to get the answer soon! and Best wishes!

Traceback (most recent call last):
File "C:\Users\Administrator\Desktop\CIAH-main\main_train.py", line 151, in
clusterer = ClusterBaseline(FLAGS.embedding_model, input_data, FLAGS.embedding_size,)
File "C:\Users\Administrator\Desktop\CIAH-main\Classifier\clusterer.py", line 23, in init
self.model = SceneMiningBaselineModel(input_data, embedding_size)
File "C:\Users\Administrator\Desktop\CIAH-main\Models\model_baseline_v1.py", line 24, in init
self.init_hypergraph_neural_network(grad_alpha)
File "C:\Users\Administrator\Desktop\CIAH-main\Models\model_baseline_v1.py", line 100, in init_hypergraph_neural_network
sub1_output = SubLayer_N2E(self.E4N, features, mask_feats,
File "C:\Users\Administrator\Desktop\CIAH-main\GNN\layer.py", line 34, in SubLayer_N2E
neighbor_concat = tf.concat(neighbor_concat, axis=-2)
File "E:\annocade\envs\pytorch\lib\site-packages\tensorflow\python\util\dispatch.py", line 206, in wrapper
return target(*args, **kwargs)
File "E:\annocade\envs\pytorch\lib\site-packages\tensorflow\python\ops\array_ops.py", line 1768, in concat
return gen_array_ops.concat_v2(values=values, axis=axis, name=name)
File "E:\annocade\envs\pytorch\lib\site-packages\tensorflow\python\ops\gen_array_ops.py", line 1212, in concat_v2
_ops.raise_from_not_ok_status(e, name)
File "E:\annocade\envs\pytorch\lib\site-packages\tensorflow\python\framework\ops.py", line 6897, in raise_from_not_ok_status