From 1c9a85d927277e76935b31eea87cc4e98b710948 Mon Sep 17 00:00:00 2001 From: Younes Strittmatter Date: Sun, 13 Oct 2024 23:38:38 -0400 Subject: [PATCH] Update __init__.py --- src/autora/experimentalist/falsification/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/autora/experimentalist/falsification/__init__.py b/src/autora/experimentalist/falsification/__init__.py index ba47336..53400bf 100644 --- a/src/autora/experimentalist/falsification/__init__.py +++ b/src/autora/experimentalist/falsification/__init__.py @@ -24,7 +24,7 @@ def pool( reference_conditions: Union[pd.DataFrame, np.ndarray], reference_observations: Union[pd.DataFrame, np.ndarray], metadata: VariableCollection, - num_samples: int = 100, + num_samples: int = 1, training_epochs: int = 1000, optimization_epochs: int = 1000, training_lr: float = 1e-3, @@ -197,7 +197,7 @@ def sample( reference_conditions: Union[pd.DataFrame, np.ndarray], reference_observations: Union[pd.DataFrame, np.ndarray], metadata: VariableCollection, - num_samples: Optional[int] = None, + num_samples: int = 1, training_epochs: int = 1000, training_lr: float = 1e-3, plot: bool = False, @@ -283,7 +283,7 @@ def falsification_score_sample( reference_conditions: Union[pd.DataFrame, np.ndarray], reference_observations: Union[pd.DataFrame, np.ndarray], metadata: Optional[VariableCollection] = None, - num_samples: Optional[int] = None, + num_samples: int = 1, training_epochs: int = 1000, training_lr: float = 1e-3, plot: bool = False, @@ -357,7 +357,7 @@ def falsification_score_sample_from_predictions( reference_conditions: Union[pd.DataFrame, np.ndarray], reference_observations: np.ndarray, metadata: Optional[VariableCollection] = None, - num_samples: Optional[int] = None, + num_samples: int = 1, training_epochs: int = 1000, training_lr: float = 1e-3, plot: bool = False,