added sentsat model

pyhealth/models/__init__.py

@@ -22,4 +22,5 @@
 from .grasp import GRASP, GRASPLayer
 from .stagenet import StageNet, StageNetLayer
 from .tcn import TCN, TCNLayer
-from .wordsat import WordSAT
+from .wordsat import WordSAT, WordSATEncoder,WordSATDecoder, WordSATAttention
+from .sentsat import SentSAT

pyhealth/models/base_model.py

@@ -22,7 +22,7 @@ class BaseModel(ABC, nn.Module):
  feature_keys: list of keys in samples to use as features,
  e.g. ["conditions", "procedures"].
  label_key: key in samples to use as label (e.g., "drugs").
- mode: one of "binary", "multiclass", or "multilabel".
+ mode: one of "binary", "multiclass", "multilabel", or sequence.
  """
 
  def __init__(
@@ -31,13 +31,16 @@ def __init__(
  feature_keys: List[str],
  label_key: str,
  mode: str,
+ save_generated_caption: bool = False
  ):
  super(BaseModel, self).__init__()
  assert mode in VALID_MODE, f"mode must be one of {VALID_MODE}"
  self.dataset = dataset
  self.feature_keys = feature_keys
  self.label_key = label_key
  self.mode = mode
+ if mode == "sequence":
+ self.save_generated_caption = save_generated_caption
  # used to query the device of the model
  self._dummy_param = nn.Parameter(torch.empty(0))
  return