diff --git a/gamestonk_terminal/prediction_techniques/fbprophet.py b/gamestonk_terminal/prediction_techniques/fbprophet.py index 3a93230555f..3f894feb7c9 100644 --- a/gamestonk_terminal/prediction_techniques/fbprophet.py +++ b/gamestonk_terminal/prediction_techniques/fbprophet.py @@ -14,6 +14,7 @@ register_matplotlib_converters() warnings.simplefilter("ignore") + def fbprophet(l_args, s_ticker, df_stock): parser = argparse.ArgumentParser( prog="fbprophet", diff --git a/gamestonk_terminal/prediction_techniques/neural_networks.py b/gamestonk_terminal/prediction_techniques/neural_networks.py index c062fd1ecf0..4fab5020788 100644 --- a/gamestonk_terminal/prediction_techniques/neural_networks.py +++ b/gamestonk_terminal/prediction_techniques/neural_networks.py @@ -8,21 +8,13 @@ from pandas.plotting import register_matplotlib_converters from TimeSeriesCrossValidation import splitTrain from sklearn.preprocessing import MinMaxScaler, StandardScaler from tensorflow.keras.models import Sequential -from tensorflow.keras.layers import ( - LSTM, - SimpleRNN, - Dense, - Dropout, - # Activation, - # RepeatVector, - # TimeDistributed, -) +from tensorflow.keras.layers import LSTM, SimpleRNN, Dense, Dropout from gamestonk_terminal.helper_funcs import ( check_positive, get_next_stock_market_days, parse_known_args_and_warn, - print_pretty_prediction + print_pretty_prediction, ) from gamestonk_terminal import config_neural_network_models as cfg_nn_models