From 3bf36ed0e91e1974ed0c6a70c846c8d5d8aca32c Mon Sep 17 00:00:00 2001 From: Manuel Eusebio de Paz Carmona Date: Thu, 9 Dec 2021 17:27:44 +0100 Subject: [PATCH] DATA: Adding data wrangling to use new Kaggle dataset --- .../RL-Bitcoin-trading-bot_1.py | 21 ++++++++++++++++++- 1 file changed, 20 insertions(+), 1 deletion(-) diff --git a/RL-Bitcoin-trading-bot_1/RL-Bitcoin-trading-bot_1.py b/RL-Bitcoin-trading-bot_1/RL-Bitcoin-trading-bot_1.py index c767e6a..5e52ab5 100644 --- a/RL-Bitcoin-trading-bot_1/RL-Bitcoin-trading-bot_1.py +++ b/RL-Bitcoin-trading-bot_1/RL-Bitcoin-trading-bot_1.py @@ -146,9 +146,28 @@ def Random_games(env, train_episodes = 50, training_batch_size=500): print('read') dirname = os.path.dirname(__file__) -filename = os.path.join(dirname, '../data/pricedata.csv') +filename = os.path.join(dirname, '../data/etcust.csv') # Kaggle +# filename = os.path.join(dirname, '../data/pricedata.csv') # reference print(f'reading file: {filename}') df = pd.read_csv(filename) + +print('turn') +df['Date'] = df['time'].apply(lambda x: pd.to_datetime(x, unit='ms').to_pydatetime().strftime("%d-%m-%Y %H:%M")) +df['Open'] = df['open'] +df['High'] = df['close'] +df['Low'] = df['high'] +df['Close'] = df['low'] +df['Volume'] = df['volume'] + +print('clean') +del df['time'] +del df['open'] +del df['close'] +del df['high'] +del df['low'] +del df['volume'] + +print('sort') df = df.sort_values('Date') print(df.head())