def _default_paired_text_dataset_hparams(): """Returns hyperparameters of a mono text dataset with default values. """ # TODO(zhiting): add more docs source_hparams = _default_mono_text_dataset_hparams() source_hparams["bos_token"] = None source_hparams["data_name"] = "source" target_hparams = _default_mono_text_dataset_hparams() target_hparams.update({ "vocab_share": False, "embedding_init_share": False, "processing_share": False, "data_name": "target" }) return {"source_dataset": source_hparams, "target_dataset": target_hparams}
def _default_paired_text_dataset_hparams(): r"""Returns hyperparameters of a paired text dataset with default values. See :meth:`texar.data.PairedTextData.default_hparams` for details. """ source_hparams = _default_mono_text_dataset_hparams() source_hparams["bos_token"] = None source_hparams["data_name"] = "source" target_hparams = _default_mono_text_dataset_hparams() target_hparams.update({ "vocab_share": False, "embedding_init_share": False, "processing_share": False, "data_name": "target" }) return {"source_dataset": source_hparams, "target_dataset": target_hparams}
def _default_dataset_hparams(data_type=None): """Returns hyperparameters of a dataset with default values. """ # TODO(zhiting): add more docs if not data_type or _is_text_data(data_type): hparams = _default_mono_text_dataset_hparams() hparams.update({ "data_type": _DataTypes.TEXT, "vocab_share_with": None, "embedding_init_share_with": None, "processing_share_with": None, }) elif _is_scalar_data(data_type): hparams = _default_scalar_dataset_hparams() return hparams
def _default_dataset_hparams(data_type=None): """Returns hyperparameters of a dataset with default values. See :meth:`texar.data.MultiAlignedData.default_hparams` for details. """ if not data_type or _is_text_data(data_type): hparams = _default_mono_text_dataset_hparams() hparams.update({ "data_type": _DataTypes.TEXT, "vocab_share_with": None, "embedding_init_share_with": None, "processing_share_with": None, }) elif _is_scalar_data(data_type): hparams = _default_scalar_dataset_hparams() return hparams