Skip to content

Commit 1e5f958

Browse files
committed
Update double_ml.py
1 parent 240e358 commit 1e5f958

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

doubleml/double_ml.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,8 +131,9 @@ def __str__(self):
131131
for key, value in self.learner.items():
132132
learner_info += f'Learner {key}: {str(value)}\n'
133133
if self.rmses is not None:
134+
learner_info += 'Out-of-sample Performance:\n'
134135
for learner in self.params_names:
135-
learner_info += f'Learner {learner} cross-fitted RMSE: {self.rmses[learner]}\n'
136+
learner_info += f'Learner {learner} RMSE: {self.rmses[learner]}\n'
136137

137138
if self._is_cluster_data:
138139
resampling_info = f'No. folds per cluster: {self._n_folds_per_cluster}\n' \

0 commit comments

Comments
 (0)