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

A Question about Low Correlations in the "stageA2_mbm_finetune" Execution on the GOD Dataset #26

Open
HuangWanqiu opened this issue Feb 21, 2024 · 5 comments

Comments

@HuangWanqiu
Copy link

Thanks for all your efforts!

When running "stageA2_mbm_finetune" with the GOD dataset, the obtained correlations are very small (approximately 0.02). When I execute "stageA2_mbm_finetune" using the following command, the correlations obtained on the GOD test dataset are extremely low, as shown in the figure below, reaching a maximum of only around 0.02. Is this a normal result?

python code/stageA2_mbm_finetune.py --dataset GOD --pretrain_mbm_path results/pretrains/GOD/fmri_encoder.pth

image

@HuangWanqiu
Copy link
Author

HuangWanqiu commented Feb 23, 2024

I found the reason. It was my mistake. After increasing the epoch number to 5000 (actually around 3000 would suffice), the correlation on the GOD test set increased to around 0.57, as shown in the figure below.

Once again, thank you for your excellent work.

image

@HuangWanqiu
Copy link
Author

I have an additional question. In your valuable experience, when adapting the code to another dataset, what would be a "good" value for the correlation (cor) of the train set/test set?
I look forward to your response.

@xyhanHIT
Copy link

xyhanHIT commented Mar 7, 2024

I found the reason. It was my mistake. After increasing the epoch number to 5000 (actually around 3000 would suffice), the correlation on the GOD test set increased to around 0.57, as shown in the figure below.

Once again, thank you for your excellent work.

image

Hello, I would like to ask if you only change the num_epoch in config.py from 15 to 5000 and maintain default values for other parameters? Look forward to your reply.

@HuangWanqiu
Copy link
Author

I found the reason. It was my mistake. After increasing the epoch number to 5000 (actually around 3000 would suffice), the correlation on the GOD test set increased to around 0.57, as shown in the figure below.
Once again, thank you for your excellent work.
image

Hello, I would like to ask if you only change the num_epoch in config.py from 15 to 5000 and maintain default values for other parameters? Look forward to your reply.

If I remember correctly, I only made very minor changes, such as adjusting the batch size and number of epochs.

@xyhanHIT
Copy link

I found the reason. It was my mistake. After increasing the epoch number to 5000 (actually around 3000 would suffice), the correlation on the GOD test set increased to around 0.57, as shown in the figure below.
Once again, thank you for your excellent work.
image

Hello, I would like to ask if you only change the num_epoch in config.py from 15 to 5000 and maintain default values for other parameters? Look forward to your reply.

If I remember correctly, I only made very minor changes, such as adjusting the batch size and number of epochs.

Okay, thank you for your reply!

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

No branches or pull requests

2 participants