search

codeKgu / Text-GCN

A PyTorch implementation of "Graph Convolutional Networks for Text Classification." (AAAI 2019)
MIT License
123 stars 25 forks source link

AssertionError #5

Open ghost opened 2 years ago

ghost commented 2 years ago

(base) C:\Users\Ramy\Downloads\Compressed\Text-GCN-master\Text-GCN-master>python main.py text_gcn: {'pred_type': 'softmax', 'node_embd': 'gcn', 'layer_dims': [10, 4], 'class_weights': True, 'dropout': True}

in _save_conf_code Logging to C:\Users\Ramy\Downloads\Compressed\Text-GCN-master\Text-GCN-master\logs\TextGNN_ag_presplit_0.8_0.1_0.1_2022-07-22T08-02-04.424720 Loaded from C:\Users\Ramy\Downloads\Compressed\Text-GCN-master\Text-GCN-master\save\split\ag_presplit_train_val_test_3_window_size_10.klepto (129620, 129620) Traceback (most recent call last): File "C:\Users\Ramy\Downloads\Compressed\Text-GCN-master\Text-GCN-master\main.py", line 44, in main() File "C:\Users\Ramy\Downloads\Compressed\Text-GCN-master\Text-GCN-master\main.py", line 20, in main saved_model, model = train(train_data, val_data, saver) File "C:\Users\Ramy\Downloads\Compressed\Text-GCN-master\Text-GCN-master\train.py", line 13, in train model = create_model(train_data) File "C:\Users\Ramy\Downloads\Compressed\Text-GCN-master\Text-GCN-master\model_factory.py", line 17, in create_model return model_ctors[name](layer_info, dataset) File "C:\Users\Ramy\Downloads\Compressed\Text-GCN-master\Text-GCN-master\model_factory.py", line 35, in create_text_gnn return TextGNN( File "C:\Users\Ramy\Downloads\Compressed\Text-GCN-master\Text-GCN-master\model_text_gnn.py", line 21, in init assert layer_dim_list[-1] == num_labels AssertionError

davut1999 commented 1 year ago

you solved it ?