Views
No views yet
1{
2 "_name_or_path": "custom-graph-tox-predictor",
3 "architectures": [
4 "GraphConvolutionalNetwork"
5 ],
6 "model_type": "molecular_property_prediction",
7 "graph_type": "molecular_graph",
8 "node_features": 74,
9 "edge_features": 12,
10 "num_gcn_layers": 3,
11 "hidden_dim": 128,
12 "global_pooling": "readout_mean",
13 "output_dim": 1,
14 "task_type": "binary_classification",
15 "id2label": {
16 "0": "Non-Toxic",
17 "1": "Toxic"
18 },
19 "label2id": {
20 "Non-Toxic": 0,
21 "Toxic": 1
22 },
23 "pytorch_version": "2.1.0"
24}