結果 : implement loss function in python