How can I edit the code in the attached photo to get 20% of the data with high accuracy involved in federated learning?