Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Minor improvement to FakeDataset #4065

Merged
merged 2 commits into from
Feb 11, 2022
Merged

Conversation

arunppsg
Copy link
Contributor

I have made a small improvement to FakeDataset class.

The node features and edge features generated have same distribution across different labels. I made a small fix to make them have different mean across different labels. This will help when the fake dataset is used for training and testing models. Earlier, since distributions of node features and edge features where same across different labels, a model could not learn effectively.

Copy link
Member

@rusty1s rusty1s left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. Thank you!

@arunppsg
Copy link
Contributor Author

arunppsg commented Feb 11, 2022

Seems I need to update the tests and also some bugs in this PR.

@arunppsg arunppsg marked this pull request as draft February 11, 2022 14:43
@arunppsg
Copy link
Contributor Author

arunppsg commented Feb 11, 2022

Some tests are failing at an assertion for data.edge_weight and data.edge_attr because the change introduced create a variance in both these features (due to dependency on graph labels during random value generation).

I am not sure how to handle it here. Can you help me here @rusty1s to get this merged in? Thanks!

@rusty1s
Copy link
Member

rusty1s commented Feb 11, 2022

How about we only offset node features?

@arunppsg
Copy link
Contributor Author

How about we only offset node features?

Sounds good. In the last update I made offset to only node features and it passes the test case. Thanks!

@arunppsg arunppsg marked this pull request as ready for review February 11, 2022 16:59
@rusty1s
Copy link
Member

rusty1s commented Feb 11, 2022

Super :)

@rusty1s rusty1s merged commit d1aa183 into pyg-team:master Feb 11, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants