image_output_1.png
import numpy as np

import pandas as pd

import torch

df = pd.read_csv('../input/mayo-clinic-strip-ai/train.csv')
df_subset = df[['patient_id']][:500]

pred = torch.tensor(np.linspace(0, 1, 500))
target = torch.ones(500, dtype=torch.float64)
df_subset['CE'], df_subset['LAA'] = [pred.numpy(), 1 - pred.numpy()]
df_subset