改造损失函数

参考https://cloud.tencent.com/developer/article/1619122
考虑到一些小样本对现实世界的影响更大（比如山洪、泥石流），采用平衡均方误差Balanced Mean Squared Error (B-MSE)和平衡平均绝对误差Balanced Mean Absolute Error (B-MAE)进行评估，他们在计算MSE和MAE时为小样本分配了更多了权重。

  • 该论文中其他的trick：TrajGRU用于处理时空相关性数据很有优势；采用马氏距离（均值加方差是否大于μ+3σ）去除噪声点；