Commit ee54de7b authored by casey's avatar casey
Browse files

Change default nb_epoch to 10

parent 1d529aac

deepchem/hyper/base_classes.py

+1 −1
Original line number Diff line number Diff line
@@ -80,7 +80,7 @@ class HyperparamOpt(object): 
                        valid_dataset: Dataset,

                        metric: Metric,

                        output_transformers: List[Transformer] = [],

                        nb_epoch: int = 5,

                        nb_epoch: int = 10,

                        use_max: bool = True,

                        logdir: Optional[str] = None,

                        **kwargs) -> Tuple[Model, Dict, Dict]:

deepchem/hyper/gaussian_process.py

+1 −1
Original line number Diff line number Diff line
@@ -132,7 +132,7 @@ class GaussianProcessHyperparamOpt(HyperparamOpt): 
                        valid_dataset: Dataset,

                        metric: Metric,

                        output_transformers: List[Transformer] = [],

                        nb_epoch: int = 5,

                        nb_epoch: int = 10,

                        use_max: bool = True,

                        logdir: Optional[str] = None,

                        max_iter: int = 20,

deepchem/hyper/grid_search.py

+1 −1
Original line number Diff line number Diff line
@@ -66,7 +66,7 @@ class GridHyperparamOpt(HyperparamOpt): 
      valid_dataset: Dataset,

      metric: Metric,

      output_transformers: List[Transformer] = [],

      nb_epoch: int = 5,

      nb_epoch: int = 10,

      use_max: bool = True,

      logdir: Optional[str] = None,

      **kwargs,

CRA Git | Maintained and supported by SUSTech CRA and CCSE