def __init__(self,
age,
alive,
features=None,
gamma=1.0,
gamma_beta=None,
bias=True,
normalize=True,
verbose=False):
"""
Initializes objects with parameters necessary to create the supporting
objects: DataHandler and ShiftedBeta
:param age: str
The column name to identify the age of each individual. Age has to
be an integer value, and will determine the time intervals the
model with work with.
--- See DataHandler.py
:param alive: str
The column name with the status of each individual. In the context
of survival analysis, an individual may be dead or alive, and its
contribution to the model will depend on it.
--- See DataHandler.py
:param features: str, list or None
A string with the name of the column to be used as features, or a
list of names of columns to be used as features or None, if no
features are to be used.
--- See DataHandler.py
:param gamma: float
A non-negative float specifying the strength of the regularization
applied to w_alpha (alpha's weights) and, if gamma_beta is not
given, it is also applied to beta.
--- See ShiftedBeta.py
:param gamma_beta: float
A non-negative float specifying the strength of the regularization
applied to w_beta (beta's weights). If specified, overwrites the
value of gamma for beta.
--- See ShiftedBeta.py
:param bias: bool
Whether or not a bias term should be added to the feature matrix.
--- See DataHandler.py
:param normalize: bool
Whether or not numerical fields should be normalized (centered and
scaled to have std=1)
--- See DataHandler.py
:param verbose: bool
Whether of not status updates should be printed
--- See ShiftedBeta.py
"""
# Create objects!
# DATA-HANDLER OBJECT
# The DataHandler object may be created without the training data, so
# we do it here.
self.dh = DataHandler(age=age,
alive=alive,
features=features,
bias=bias,
normalize=normalize)
# Shifted beta model object
# Was a different gammab parameter passed? If not, we use the same
# value passed to gamma.
if gamma_beta is None:
gamma_beta = 1.0 * gamma
# create shifted beta object
self.sb = ShiftedBetaGeometric(gamma_alpha=gamma,
gamma_beta=gamma_beta,
verbose=verbose)
