dt_mxnet_part2_line_chart_2.py

# lineral equation
def f2(x):
    # a = 5
    # b = 2
    return - 5 * x + 2

# Data
X2 = np.arange(100, step=0.001)
Y2 = f2(X2)

# Split data for taining and evaluation
X2_train, X2_test, Y2_train, Y2_test = train_test_split(X2, Y2)
figure = plt.figure()
plt.plot(X2_train, Y2_train, 'r')
plt.xlabel('x')
plt.ylabel('y')
plt.title('simple line')
plt.show()