yalaudah / facies_classification_benchmark

The repository includes PyTorch code, and the data, to reproduce the results for our paper titled "A Machine Learning Benchmark for Facies Classification" (published in the SEG Interpretation Journal, August 2019).
MIT License
113 stars 62 forks source link

Issue in section_test.py #16

Closed motazalfarraj closed 2 years ago

motazalfarraj commented 2 years ago

In section test, line#28:

splits = [args.split if 'both' not in args.split else 'test1', 'test2']

when args.split = "test2", the output split will be ["test2", "test2"] when args.split = "test1", the output split will be ["test1", "test2"]

@yalaudah

yalaudah commented 2 years ago

Thanks! that's interesting that I didn't pick this up before. I think it's missing extra brackets for ['test1', 'test2'].

yalaudah commented 2 years ago

@motazalfarraj I'm closing this issue as it has been fixed in PR #17. Thanks again for pointing this out!