docstr: Fits the CEVAE model and then predicts.
  
  Args:
      X (np.matrix or np.array or pd.Dataframe): a feature matrix
      treatment (np.array or pd.Series): a treatment vector
      y (np.array or pd.Series): an outcome vector
  Returns:
      (np.ndarray): Predictions of treatment effects.
docstr_tok: ["Fits","the","CEVAE","model","and","then","predicts",".","Args",":","X","(","np.matrix","or","np.array","or","pd.Dataframe",")",":","a","feature","matrix","treatment","(","np.array","or","pd.Series",")",":","a","treatment","vector","y","(","np.array","or","pd.Series",")",":","an","outcome","vector","Returns",":","(","np.ndarray",")",":","Predictions","of","treatment","effects","."]
code: def fit_predict(self, X, treatment, y, p=None):
      """
      Fits the CEVAE model and then predicts.
  
      Args:
          X (np.matrix or np.array or pd.Dataframe): a feature matrix
          treatment (np.array or pd.Series): a treatment vector
          y (np.array or pd.Series): an outcome vector
      Returns:
          (np.ndarray): Predictions of treatment effects.
      """
      self.fit(X, treatment, y)
      return self.predict(X)
  
file_path: causalml/causalml/inference/nn/cevae.py
imports: import logging
  import torch
  from pyro.contrib.cevae import CEVAE
  from causalml.inference.meta.utils import convert_pd_to_np
label_desc: Use image node_id 4 for calling the CEVAE obj's underlying member method code with example usage: obj.fit_predict(X, treatment, y, p) and returns: self