Skip to content

Commit

Permalink
Update phenotyping.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Chirag Nagpal authored Apr 7, 2022
1 parent 67d106d commit f3ad9fe
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions auton_survival/phenotyping.py
Original file line number Diff line number Diff line change
Expand Up @@ -326,12 +326,8 @@ def fit(self, features):
"""

if self.dim_red_method is not None:
<<<<<<< HEAD
print("Fitting the following Dimensionality Reduction Model:\n",
self.dim_red_model)
=======
print("Fitting the following Dimensionality Reduction Model:\n", self.dim_red_model)
>>>>>>> db17080dcfd91df1c07937d54c479ae6ab1eaf88
self.dim_red_model = self.dim_red_model.fit(features)
features = self.dim_red_model.transform(features)

Expand Down Expand Up @@ -471,10 +467,7 @@ def fit(self, features, outcomes, interventions, horizon):

times = np.unique(outcomes.times.values)
cf_predictions = self.cf_model.predict_counterfactual_survival(features,
<<<<<<< HEAD
=======
interventions,
>>>>>>> db17080dcfd91df1c07937d54c479ae6ab1eaf88
times)

ite_estimates = cf_predictions[1] - cf_predictions[0]
Expand Down

0 comments on commit f3ad9fe

Please sign in to comment.