Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- labels=parsed['pred']
- #kdj_gold = tf.reshape(parsed['KDJ_gold'],shape=[3,0])
- kdj_gold=tf.cast(parsed['KDJ_gold'],tf.string)
- return {'volume':parsed['volume'],
- 'Quote_asset_volume': parsed['Quote_asset_volume'],
- 'Number_of_trades': parsed['Number_of_trades'],
- 'Taker_buy_base_asset_volume': parsed['Taker_buy_base_asset_volume'],
- 'Taker_buy_quote_asset_volume': parsed['Taker_buy_quote_asset_volume'],
- '5day_press': parsed['5day_press'],
- '30day_press': parsed['30day_press'],
- '60day_press': parsed['60day_press'],
- 'change': parsed['change'],
- 'change_day': parsed['change_day'],
- 'day_ofyear': parsed['day_ofyear'],
- 'series': parsed['series'],
- 'num_updowm': parsed['num_updowm'],
- 'a': parsed['a'],
- 'up_shaow_day': parsed['up_shaow_day'],
- 'down_shaow_day': parsed['down_shaow_day'],
- 'up_shaow_15min': parsed['up_shaow_15min'],
- 'down_shaow_15min': parsed['down_shaow_15min'],
- 'szzs': parsed['szzs'],
- 'dif': parsed['dif'],
- 'dea': parsed['dea'],
- 'macd': parsed['macd'],
- 'K': parsed['K'],
- 'D': parsed['D'],
- 'J': parsed['J'],
- 'KDJ_gold': kdj_gold,
- 'low_rate': parsed['low_rate'],
- 'high_rate': parsed['high_rate'],
- 'time': parsed['time'],
- 'time_next': parsed['time_next'],
- 'RSI': parsed['RSI'],
- },labels
- parsed_dataset = raw_dataset.map(_parse_function)
- parsed_dataset= parsed_dataset.repeat()
- parsed_dataset= parsed_dataset.batch(batch_size)#n_step)
- parsed_dataset = parsed_dataset.window(5)
- print(parsed_dataset.output_shapes)
- return parsed_dataset
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement