I improved my railroad dataset to make memberships work a lot better. Check it out! It’s on GitHub and HuggingFace.