Skip to content

Commit

Permalink
better nomo plots
Browse files Browse the repository at this point in the history
  • Loading branch information
austinorr committed Jun 15, 2023
1 parent e8ad7ac commit cd37e4c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions nereid/nereid/src/nomograph/interpolators.py
Original file line number Diff line number Diff line change
Expand Up @@ -312,7 +312,7 @@ def _basesurface(self, ax: Optional[Axes] = None, **kwargs: Dict[str, Any]) -> A
ax.tricontourf(self.x_data, self.t_data, self.y_data, levels=255)

t = numpy.linspace(1, numpy.max(self.t_data), 100)
for i, perf in enumerate([0.6, 0.8, 0.9, 0.95, 0.97]):
for i, perf in enumerate([0.2, 0.4, 0.6, 0.8, 0.9, 0.95, 0.97]):
x = [self(t=_t, y=perf) for _t in t]
ax.plot(x, t, c=f"C{i}", label=f"{perf:.0%}")

Expand Down Expand Up @@ -359,7 +359,7 @@ def plot(self, *args: Tuple, **kwargs: Dict[str, Any]) -> Axes:
ax = self.nomo._baseplot(*args, **kwargs) # type: ignore
ax.set_xlabel("size")
ax.set_ylabel("performance")
ax.legend(ncol=2, title="ddt")
ax.legend(loc=6, bbox_to_anchor=(1.01, 0.5), ncol=2, title="ddt")
return ax

def surface_plot(self, *args: Tuple, **kwargs: Dict[str, Any]) -> Axes:
Expand Down

0 comments on commit cd37e4c

Please sign in to comment.