【发布时间】:2021-05-20 10:25:55
【问题描述】:
这只是代码的一部分。它运行但它永远不会完成或停止。这只是一个持续的过程。谁能看到我哪里出错了?
我相信这可能与数据重塑和对重塑数据的 svr 分析有关,因为我尝试通过注释行并查看错误来进行手动故障排除。一切似乎都在运行,直到到达 'lin_svr'。
代码使用数据集并创建两个 numpy 数组:“index_list”和“adj_close_prices”。指数和收盘价分别附加到这些数组中,然后代码尝试绘制指数与收盘价的对比。这就是代码的目的。
您能指出代码中的错误可能在哪里吗?
index_list = np.array([])
adj_close_prices = np.array([])
##get the index and adjusted close prices
df_days = df.loc[:, 'index']
df_adj_close = df.loc[:, 'Adj Close']
for day in df_days:
index_list = np.append(index_list, float(day))
for adj_close_price in df_adj_close:
adj_close_prices = np.append(adj_close_prices, float(adj_close_price))
index_list = index_list.reshape(-1, 1)
lin_svr = SVR(kernel='linear', C=1000.0)
lin_svr.fit(index_list, adj_close_prices)
poly_svr = SVR(kernel='poly', C=1000.0, degree=2)
poly_svr.fit(index_list, adj_close_prices)
rbf_svr = SVR(kernel='rbf', C=1000.0, gamma=0.15)
rbf_svr.fit(index_list, adj_close_prices)
##plot
plt.figure(figsize=(16, 8))
plt.scatter(index_list, adj_close_prices, color='red', label='Data')
plt.plot(index_list, rbf_svr.predict(index_list), color='green', label='RBF Model')
plt.plot(index_list, poly_svr.predict(index_list), color='orange', label='Polynomial Model')
plt.plot(index_list, lin_svr.predict(index_list), color='blue', label='Linear Model')
plt.legend()
plt.show()
【问题讨论】:
标签: python machine-learning svm analysis stock