Skip to content

Commit

Permalink
report in notebook (#298)
Browse files Browse the repository at this point in the history
  • Loading branch information
pplonski committed Jan 15, 2021
1 parent f6ef8b5 commit 317a2f4
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions supervised/base_automl.py
Original file line number Diff line number Diff line change
Expand Up @@ -1003,12 +1003,12 @@ def select_and_save_best(self):
ldb.to_csv(os.path.join(self._results_path, "leaderboard.csv"), index=False)

# save report
ldb["Link"] = [f"[Results link]({m}/README.md)" for m in ldb["name"].values]
ldb.insert(loc=0, column="Best model", value="")
ldb.loc[
ldb.name == self._best_model.get_name(), "Best model"
] = "**the best**"

ldb["name"] = [f"[{m}]({m}/README.md)" for m in ldb["name"].values]

with open(os.path.join(self._results_path, "README.md"), "w") as fout:
fout.write(f"# AutoML Leaderboard\n\n")
fout.write(tabulate(ldb.values, ldb.columns, tablefmt="pipe"))
Expand Down Expand Up @@ -1622,7 +1622,6 @@ def _md_to_html(self, md_fname, page_type):
return html_fname

def _report(self, width=900, height=1200):
print("Report from", self._results_path)

from IPython.display import IFrame

Expand Down

0 comments on commit 317a2f4

Please sign in to comment.