working on improving train vs. test split in gensgd