jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-11-05 17:01:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix black complaints

Browse Source
This commit is contained in:
didier
2021-03-13 23:15:26 +00:00
parent 2602e9ca64
commit 1e6ab04d5d
2 changed files with 24 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
gamestonk_terminal/prediction_techniques/neural_networks.py
View File

@@ -186,12 +186,10 @@ def mlp(l_args, s_ticker, df_stock):
)
return
if (
ns_parser.s_end_date
< get_next_stock_market_days(
last_stock_day=df_stock.index[0], n_next_days=ns_parser.n_inputs + ns_parser.n_days
)[-1]
):
if ns_parser.s_end_date < get_next_stock_market_days(
last_stock_day=df_stock.index[0],
n_next_days=ns_parser.n_inputs + ns_parser.n_days,
)[-1]:
print(
"Backtesting not allowed, since End Date is too close to Start Date to train model\n"
)
@@ -561,12 +559,10 @@ def rnn(l_args, s_ticker, df_stock):
)
return
if (
ns_parser.s_end_date
< get_next_stock_market_days(
last_stock_day=df_stock.index[0], n_next_days=ns_parser.n_inputs + ns_parser.n_days
)[-1]
):
if ns_parser.s_end_date < get_next_stock_market_days(
last_stock_day=df_stock.index[0],
n_next_days=ns_parser.n_inputs + ns_parser.n_days,
)[-1]:
print(
"Backtesting not allowed, since End Date is too close to Start Date to train model\n"
)
@@ -936,12 +932,10 @@ def lstm(l_args, s_ticker, df_stock):
)
return
if (
ns_parser.s_end_date
< get_next_stock_market_days(
last_stock_day=df_stock.index[0], n_next_days=ns_parser.n_inputs + ns_parser.n_days
)[-1]
):
if ns_parser.s_end_date < get_next_stock_market_days(
last_stock_day=df_stock.index[0],
n_next_days=ns_parser.n_inputs + ns_parser.n_days,
)[-1]:
print(
"Backtesting not allowed, since End Date is too close to Start Date to train model\n"
)

20
gamestonk_terminal/prediction_techniques/regression.py
View File

@@ -106,12 +106,10 @@ def regression(l_args, s_ticker, df_stock, polynomial):
)
return
if (
ns_parser.s_end_date
< get_next_stock_market_days(
last_stock_day=df_stock.index[0], n_next_days=ns_parser.n_inputs + ns_parser.n_days
)[-1]
):
if ns_parser.s_end_date < get_next_stock_market_days(
last_stock_day=df_stock.index[0],
n_next_days=ns_parser.n_inputs + ns_parser.n_days,
)[-1]:
print(
"Backtesting not allowed, since End Date is too close to Start Date to train model\n"
)
@@ -260,7 +258,10 @@ def regression(l_args, s_ticker, df_stock, polynomial):
)
plt.title("BACKTESTING: Real data price versus Prediction")
plt.xlim(df_stock.index[-1], df_pred.index[-1] + datetime.timedelta(days=1))
plt.xticks([df_stock.index[-1], df_pred.index[-1]+datetime.timedelta(days=1)], visible=True)
plt.xticks(
[df_stock.index[-1], df_pred.index[-1] + datetime.timedelta(days=1)],
visible=True,
)
plt.ylabel("Share Price ($)")
plt.grid(b=True, which="major", color="#666666", linestyle="-")
plt.minorticks_on()
@@ -300,7 +301,10 @@ def regression(l_args, s_ticker, df_stock, polynomial):
c="red",
)
plt.xlim(df_stock.index[-1], df_pred.index[-1] + datetime.timedelta(days=1))
plt.xticks([df_stock.index[-1], df_pred.index[-1]+datetime.timedelta(days=1)], visible=True)
plt.xticks(
[df_stock.index[-1], df_pred.index[-1] + datetime.timedelta(days=1)],
visible=True,
)
plt.xlabel("Time")
plt.ylabel("Prediction Error (%)")
plt.grid(b=True, which="major", color="#666666", linestyle="-")