ConsistencyEnsemble(consistency_transform=PairwiseConsistency(include_diagonal=True,\n", + " indeterminacy=Linear()),\n", + " estimator=CEBRA(batch_size=512, conditional='time',\n", + " max_iterations=500,\n", + " model_architecture='offset10-model',\n", + " output_dimension=3, temperature=1.12,\n", + " time_offsets=10, verbose=True))In a Jupyter environment, please rerun this cell to show the HTML representation or trust the notebook.
| \n", + " | estimator | \n", + "CEBRA(batch_s... verbose=True) | \n", + "
| \n", + " | consistency_transform | \n", + "PairwiseConsi...nacy=Linear()) | \n", + "
| \n", + " | random_states | \n", + "5 | \n", + "
PairwiseConsistency(include_diagonal=True, indeterminacy=Linear())
Linear()
| \n", + " | copy_X | \n", + "True | \n", + "
| \n", + " | n_jobs | \n", + "None | \n", + "
| \n", + " | positive | \n", + "False | \n", + "
CEBRA(batch_size=512, conditional='time', max_iterations=500,\n", + " model_architecture='offset10-model', output_dimension=3, temperature=1.12,\n", + " time_offsets=10, verbose=True)
| \n", + " | model_architecture | \n", + "'offset10-model' | \n", + "
| \n", + " | device | \n", + "'cuda_if_available' | \n", + "
| \n", + " | criterion | \n", + "'infonce' | \n", + "
| \n", + " | distance | \n", + "'cosine' | \n", + "
| \n", + " | conditional | \n", + "'time' | \n", + "
| \n", + " | temperature | \n", + "1.12 | \n", + "
| \n", + " | temperature_mode | \n", + "'constant' | \n", + "
| \n", + " | min_temperature | \n", + "0.1 | \n", + "
| \n", + " | time_offsets | \n", + "10 | \n", + "
| \n", + " | delta | \n", + "None | \n", + "
| \n", + " | max_iterations | \n", + "500 | \n", + "
| \n", + " | max_adapt_iterations | \n", + "500 | \n", + "
| \n", + " | batch_size | \n", + "512 | \n", + "
| \n", + " | learning_rate | \n", + "0.0003 | \n", + "
| \n", + " | optimizer | \n", + "'adam' | \n", + "
| \n", + " | output_dimension | \n", + "3 | \n", + "
| \n", + " | verbose | \n", + "True | \n", + "
| \n", + " | num_hidden_units | \n", + "32 | \n", + "
| \n", + " | pad_before_transform | \n", + "True | \n", + "
| \n", + " | hybrid | \n", + "False | \n", + "
| \n", + " | optimizer_kwargs | \n", + "(('betas', ...), ...) | \n", + "
| \n", + " | masking_kwargs | \n", + "None | \n", + "