report updated
This commit is contained in:
parent
e1b2b68f26
commit
2263b1ab7d
|
@ -3,27 +3,10 @@ import statistics as stats
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
|
|
||||||
def _in_div(s):
|
|
||||||
return "<div>" + s + "</div>\n"
|
|
||||||
|
|
||||||
def _header_footer(s):
|
def _fmt_line(s):
|
||||||
return (
|
return f"> {s} \n"
|
||||||
"""
|
|
||||||
<html>
|
|
||||||
<head>
|
|
||||||
<style>
|
|
||||||
.dataframe {
|
|
||||||
tr:hover {
|
|
||||||
background-color: aquamarine;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
</style>
|
|
||||||
</head>
|
|
||||||
<body>
|
|
||||||
""" +
|
|
||||||
s +
|
|
||||||
"</body></html>"
|
|
||||||
)
|
|
||||||
|
|
||||||
class EvaluationReport:
|
class EvaluationReport:
|
||||||
def __init__(self, prefix=None):
|
def __init__(self, prefix=None):
|
||||||
|
@ -139,18 +122,17 @@ class EvaluationReport:
|
||||||
def target(self, val):
|
def target(self, val):
|
||||||
self._target = val
|
self._target = val
|
||||||
|
|
||||||
def to_html(self, *metrics):
|
def to_md(self, *metrics):
|
||||||
res = _in_div("target: " + self.target)
|
res = _fmt_line("target: " + self.target)
|
||||||
for k, v in self.prevs.items():
|
for k, v in self.prevs.items():
|
||||||
res += _in_div(f"{k}: {str(v)}")
|
res += _fmt_line(f"{k}: {str(v)}")
|
||||||
for k, v in self.times.items():
|
for k, v in self.times.items():
|
||||||
res += _in_div(f"{k}: {v:.3f}s")
|
res += _fmt_line(f"{k}: {v:.3f}s")
|
||||||
res += "\n"
|
res += "\n"
|
||||||
for m in metrics:
|
for m in metrics:
|
||||||
res += self.get_dataframe(metrics=m).to_html() + "\n\n"
|
res += self.get_dataframe(metrics=m).to_html() + "\n\n"
|
||||||
|
|
||||||
return _header_footer(res)
|
return res
|
||||||
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def combine_reports(*args):
|
def combine_reports(*args):
|
||||||
|
|
|
@ -25,7 +25,7 @@ def estimate_comparison():
|
||||||
er = method.evaluate_comparison(DATASET, target=target)
|
er = method.evaluate_comparison(DATASET, target=target)
|
||||||
er.target = target
|
er.target = target
|
||||||
with open(OUTPUT_FILE, "a") as f:
|
with open(OUTPUT_FILE, "a") as f:
|
||||||
f.write(er.to_html(["acc"], ["f1"]))
|
f.write(er.to_md(["acc"], ["f1"]))
|
||||||
except Exception:
|
except Exception:
|
||||||
traceback.print_exc()
|
traceback.print_exc()
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue