Keras Implementation of the Code

[narminGhaffari]

Thank you for sharing this code.
I am trying to implement this code. However, I have the main question regarding:

lbl_pred = model(graph, masked_cls)

In this line of code, the graph contains the images and maskled_cls is the cluster labels, right?
the output of this model, is binary values of 0 and 1, indicating the event value or it is the survival time?

[eballot21]

Dear Narmin Ghaffari,

I am trying to implement the code too in my slides but I got an error :
The object named 'graph' is a torch.cuda.DoubleTensor and the object masked_cls is a torch.cuda.FloatTensor and should be in the same format.

Yao Jiawen advised me to make it as the float32 tensor.

Could you please confirm that the expected format for the "graph" object is a Float Tensor?

Thank you in advance
Elise