src.preprocessing.impute_dates

Impute missing dates by filling them with the most similar embedding.

Classes

ImputeDates()

Impute missing dates.
class src.preprocessing.impute_dates.ImputeDates[source]

Impute missing dates.

Replaces NaN values in the date column with one from the closest document embedding.

custom_transform(data, **transform_args)[source]

Set the missing dates to the fill_date.

Parameters:

  • data (DataFrame) – The data to transform.

  • transform_args (Never) – [UNUSED] Additional keyword arguments.

Return type:

DataFrame

Returns:

The transformed data.