deepcure package

deepcure is a python package for survival cure models based on TensorFlow/Keras.

Installation

This package is in its early development stage.

To install the development version from GitHub:

pip install git+https://github.com/vhmedina/deepcure.git

Usage

from deepcure.models import DeepPTM
from tensorflow.keras.layer import Dense

stack_eta = [Dense(1, activation='linear', name = "eta")] # Definition of the stack of layers for eta in promotion time cure model
model = DeepPTM(stack_eta = stack_eta, t_func='pe', break_val=break_val) 
model.compile(optimizer='adam')
model.fit(x_train, y_train, epochs=10)

See the tutorials for more details and features.