minor work on train and predict

Created by  quietlight  on March 5, 2024
RQ56K2G4ZCSZLOXCAPZFWAPW7PC7KTGJXUYD6MFOK4P6DQ2JYKGAC
Change contents