ConditionalSampler.fit

ConditionalSampler.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