Figure.savefig(ファイル名) で画像として保存すればいいのではないでしょうか。
matplotlib.pyplot.savefig — Matplotlib 3.1.2 documentation
python
1import numpy as np
2from matplotlib import pyplot as plt
3
4
5def relu(x):
6 return np.maximum(0, x)
7
8
9def sigmoid(x):
10 return 1 / (1 + np.exp(-x))
11
12
13def savefig(x, y, name):
14 fig, ax = plt.subplots()
15 ax.plot(x, y)
16 ax.grid()
17 fig.savefig(f"{name}.png")
18
19
20# ReLU
21x = np.linspace(-5, 5)
22y = relu(x)
23savefig(x, y, "relu")
24
25# sigmoid
26x = np.linspace(-5, 5)
27y = sigmoid(x)
28savefig(x, y, "sigmoid")