DataImputer.fit

DataImputer.fit(df_train, num_iterations, callback=None, callback_freq=0)

Train the generative model for the given iterations.

Repeated calls continue training the model, possibly on different data.

Parameters

• df_train (DataFrame) – The training data.

• num_iterations (Optional[int]) – The number of training iterations (not epochs).

• callback (Optional[Callable[[object, int, dict], bool]]) – A callback function, e.g. for logging purposes. Takes the synthesizer instance, the iteration number, and a dictionary of values (usually the losses) as arguments. Aborts training if the return value is True.

• callback_freq (int) – Callback frequency.

Return type

None

synthesized.DataImputer DataImputer.impute_mask