stadlmax / Graph-Posterior-Network

Graph Posterior Network: Bayesian Predictive Uncertainty for Node Classification (NeurIPS 2021)
https://www.daml.in.tum.de/graph-postnet
MIT License
39 stars 12 forks source link

Test set proportion in data/split.py #5

Open fyj-bit opened 1 year ago

fyj-bit commented 1 year ago

Upon analyzing the sample_per_class function in the code (around lines 188 and 189), I noticed that it seems to be causing a discrepancy in the size of the test set. According to the code logic, it results in the test set being approximately (1-20%) * 80% of the total data, rather than the stated 80% of the total data.