cardea.problem_definition.MortalityPrediction.generate_cutoff_times

MortalityPrediction.generate_cutoff_times(es)[source]

Generates cutoff times for the prediction problem.

Parameters

es (featuretools.EntitySet) – An EntitySet with the loaded data.

Returns

  • An updated EntitySet if a new column is generated.

  • A string indicating the selected target entity.

  • A dataframe of cutoff times and their target labels.

Return type

featuretools.EntitySet, str, pandas.DataFrame

Raises

ValueError – An error occurs if the cutoff variable does not exist.