INNER CODE UNIT · Python
end_date
RichardS0268/CNN-for-Trading · inference.py:22
end_date = sub_points[m_idx+1], \
mode = 'inference', \
label = setting.TRAIN.LABEL, \
indicators = setting.DATASET.INDICATORS, \
show_volume = setting.DATASET.SHOW_VOLUME, \
parallel_num=setting.DATASET.PARALLEL_NUM)
inference_imageset = inference_dataset.generate_images(1.0)
for id in range(len(inference_imageset)-1):
if len(inference_dataset[id][1]) == 0:
continue
inference_imgs = []
for img in inference_dataset[id][1]:
inference_imgs.append(img[0])
input = torch.Tensor(np.array(inference_imgs))
input = input.to(device)
output = model(input)[:, 1]
up_factors = []