Transformer进行船舶轨迹预测 完整代码数据

项目讲解：Transformer进行船舶轨迹预测 -完整代码数据视频简介链接自取_哔哩哔哩_bilibili

本博客付完整的代码数据

项目代码：

import argparse
from dataread import data_train,data_test
import torch
import numpy as np
from transformer.Models import Transformer
import os
from torch import optim
from transformer.Optim import ScheduledOptim
from torch.nn import functional as F
import random
from tqdm import tqdm
from matplotlib import pyplot as plt
os.environ['CUDA_LAUNCH_BLOCKING'] ='1'

def DrawTrajectory(tra_pred,tra_true):
    tra_pred[:,:,0]=tra_pred[:,:,0]*0.00063212+110.12347
    tra_true[:,:,0]=tra_true[:,:,0]*0.00063212+110.12347