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