compute_ntk_data_score.py