class_docstr: Coref model for a single edge. Compatible with EdgeFeaturesDataset.
class_docstr_tok: ["Coref","model","for","a","single","edge",".","Compatible","with","EdgeFeaturesDataset","."]
docstr: Save model weights.
  
  To re-load, pass the path to the constructor instead of the name of a
  base model.
  
  Args:
    path: directory to save to. Will write several files here.
docstr_tok: ["Save","model","weights",".","To","re-load",",","pass","the","path","to","the","constructor","instead","of","the","name","of","a","base","model",".","Args",":","path",":","directory","to","save","to",".","Will","write","several","files","here","."]
code: def save(self, path: str):
      """Save model weights.
  
      To re-load, pass the path to the constructor instead of the name of a
      base model.
  
      Args:
        path: directory to save to. Will write several files here.
      """
      if not os.path.isdir(path):
          os.mkdir(path)
      self.model.save(path)
  
imports: import os
  from collections.abc import Iterable
  from typing import Optional
  from absl import logging
  from lit_nlp.api import dataset
  from lit_nlp.api import dtypes
  from lit_nlp.api import model
  from lit_nlp.api import types
  from lit_nlp.lib import utils
  import numpy
  import tensorflow
label_desc: Use image node_id 6 for calling the SingleEdgePredictor obj's underlying member method code with example usage: obj.save(path) without return types