复制链接

克隆策略

In [6]:

# 本代码由可视化策略环境自动生成 2019年4月3日 13:48
# 本代码单元只能在可视化模式下编辑。您也可以拷贝代码，粘贴到新建的代码单元或者策略，然后修改。


# Python 代码入口函数，input_1/2/3 对应三个输入端，data_1/2/3 对应三个输出端
def m23_run_bigquant_run(input_1, input_2, input_3):
    fields = ['open','high','low','close','volume']
    input_1_df = input_1.read_pickle()
    ins = input_1_df['instruments']
    start_date = input_1_df['start_date']
    end_date = input_1_df['end_date']
    df = D.history_data(ins, start_date, end_date, fields)     
    data_1 = DataSource.write_df(df)
    return Outputs(data_1=data_1, data_2=None, data_3=None)

# 后处理函数，可选。输入是主函数的输出，可以在这里对数据做处理，或者返回更友好的outputs数据格式。此函数输出不会被缓存。
def m23_post_run_bigquant_run(outputs):
    return outputs

# Python 代码入口函数，input_1/2/3 对应三个输入端，data_1/2/3 对应三个输出端
def m16_run_bigquant_run(input_1, input_2, input_3):
    input_ds = input_1
    df = input_ds.read_df()
    df['return'] = (df.close.shift(-10)/df.close - 1)
    df['label'] = np.where(df['return'] > 0, 1, 0)
    ds = DataSource.write_df(df)
    return Outputs(data_1=ds)


# 后处理函数，可选。输入是主函数的输出，可以在这里对数据做处理，或者返回更友好的outputs数据格式。此函数输出不会被缓存。
def m16_post_run_bigquant_run(outputs):
    return outputs

# Python 代码入口函数，input_1/2/3 对应三个输入端，data_1/2/3 对应三个输出端
def m2_run_bigquant_run(input_1, input_2, input_3):
    input_series = input_1
    input_df = input_2
    test_data = input_df.read_pickle()
    pred_label = input_series.read_pickle()
  
    pred_result = pred_label.reshape(pred_label.shape[0]) 
    dt = input_3.read_df()['date'][-1*len(pred_result):]
    pred_df = pd.Series(pred_result, index=dt)
    ds = DataSource.write_df(pred_df)
    
    pred_label = np.where(pred_label>0.5,1,0)
    labels = test_data['y']
    print('准确率%s'%(np.mean(pred_label==labels)))
    
    return Outputs(data_1=ds)

# 后处理函数，可选。输入是主函数的输出，可以在这里对数据做处理，或者返回更友好的outputs数据格式。此函数输出不会被缓存。
def m2_post_run_bigquant_run(outputs):
    return outputs

# 回测引擎：每日数据处理函数，每天执行一次
def m1_handle_data_bigquant_run(context, data):
    # 按日期过滤得到今日的预测数据
    try:
        prediction = context.prediction[data.current_dt.strftime('%Y-%m-%d')]
    except KeyError as e:
        return
    
    instrument = context.instruments[0]
    sid = context.symbol(instrument)
    cur_position = context.portfolio.positions[sid].amount
    
    # 交易逻辑
    if prediction > 0.5 and cur_position == 0:
        context.order_target_percent(context.symbol(instrument), 1)
        print(data.current_dt, '买入！')
        
    elif prediction < 0.5 and cur_position > 0:
        context.order_target_percent(context.symbol(instrument), 0)
        print(data.current_dt, '卖出！')
    
# 回测引擎：准备数据，只执行一次
def m1_prepare_bigquant_run(context):
    pass

# 回测引擎：初始化函数，只执行一次
def m1_initialize_bigquant_run(context):
    # 加载预测数据
    context.prediction = context.options['data'].read_df()

    # 系统已经设置了默认的交易手续费和滑点，要修改手续费可使用如下函数
    context.set_commission(PerOrder(buy_cost=0.0003, sell_cost=0.0013, min_cost=5))
# 回测引擎：每个单位时间开始前调用一次，即每日开盘前调用一次。
def m1_before_trading_start_bigquant_run(context, data):
    pass


m3 = M.dl_layer_input.v1(
    shape='100,1',
    batch_shape='',
    dtype='float32',
    sparse=False,
    name=''
)

m4 = M.dl_layer_lstm.v1(
    inputs=m3.data,
    units=100,
    activation='tanh',
    recurrent_activation='hard_sigmoid',
    use_bias=True,
    kernel_initializer='glorot_uniform',
    recurrent_initializer='Orthogonal',
    bias_initializer='Ones',
    unit_forget_bias=True,
    kernel_regularizer='None',
    kernel_regularizer_l1=0,
    kernel_regularizer_l2=0,
    recurrent_regularizer='None',
    recurrent_regularizer_l1=0,
    recurrent_regularizer_l2=0,
    bias_regularizer='None',
    bias_regularizer_l1=0,
    bias_regularizer_l2=0,
    activity_regularizer='None',
    activity_regularizer_l1=0,
    activity_regularizer_l2=0,
    kernel_constraint='None',
    recurrent_constraint='None',
    bias_constraint='None',
    dropout=0,
    recurrent_dropout=0,
    return_sequences=True,
    implementation='0',
    name=''
)

m10 = M.dl_layer_dropout.v1(
    inputs=m4.data,
    rate=0.9,
    noise_shape='',
    name=''
)

m25 = M.dl_layer_lstm.v1(
    inputs=m10.data,
    units=100,
    activation='tanh',
    recurrent_activation='hard_sigmoid',
    use_bias=True,
    kernel_initializer='glorot_uniform',
    recurrent_initializer='Orthogonal',
    bias_initializer='Zeros',
    unit_forget_bias=True,
    kernel_regularizer='None',
    kernel_regularizer_l1=0,
    kernel_regularizer_l2=0,
    recurrent_regularizer='None',
    recurrent_regularizer_l1=0,
    recurrent_regularizer_l2=0,
    bias_regularizer='None',
    bias_regularizer_l1=0,
    bias_regularizer_l2=0,
    activity_regularizer='None',
    activity_regularizer_l1=0,
    activity_regularizer_l2=0,
    kernel_constraint='None',
    recurrent_constraint='None',
    bias_constraint='None',
    dropout=0,
    recurrent_dropout=0,
    return_sequences=False,
    implementation='0',
    name=''
)

m11 = M.dl_layer_dropout.v1(
    inputs=m25.data,
    rate=0.9,
    noise_shape='',
    name=''
)

m9 = M.dl_layer_dense.v1(
    inputs=m11.data,
    units=1,
    activation='sigmoid',
    use_bias=True,
    kernel_initializer='glorot_uniform',
    bias_initializer='Zeros',
    kernel_regularizer='None',
    kernel_regularizer_l1=0,
    kernel_regularizer_l2=0,
    bias_regularizer='None',
    bias_regularizer_l1=0,
    bias_regularizer_l2=0,
    activity_regularizer='None',
    activity_regularizer_l1=0,
    activity_regularizer_l2=0,
    kernel_constraint='None',
    bias_constraint='None',
    name=''
)

m12 = M.dl_layer_activation.v1(
    inputs=m9.data,
    activation='tanh',
    name=''
)

m5 = M.dl_model_init.v1(
    inputs=m3.data,
    outputs=m12.data
)

m8 = M.input_features.v1(
    features='shift(close,1)'
)

m24 = M.instruments.v2(
    start_date='2015-01-01',
    end_date='2018-10-30',
    market='CN_STOCK_A',
    instrument_list='600009.SHA',
    max_count=0
)

m23 = M.cached.v3(
    input_1=m24.data,
    run=m23_run_bigquant_run,
    post_run=m23_post_run_bigquant_run,
    input_ports='',
    params='{}',
    output_ports=''
)

m16 = M.cached.v3(
    input_1=m23.data_1,
    run=m16_run_bigquant_run,
    post_run=m16_post_run_bigquant_run,
    input_ports='',
    params='{}',
    output_ports=''
)

m26 = M.derived_feature_extractor.v3(
    input_data=m23.data_1,
    features=m8.data,
    date_col='date',
    instrument_col='instrument',
    drop_na=False,
    remove_extra_columns=False,
    user_functions={}
)

m17 = M.join.v3(
    data1=m16.data_1,
    data2=m26.data,
    on='date',
    how='inner',
    sort=True
)

m18 = M.dropnan.v1(
    input_data=m17.data
)

m19 = M.filter.v3(
    input_data=m18.data,
    expr='date<\'2017-03-01\'',
    output_left_data=False
)

m13 = M.dl_convert_to_bin.v2(
    input_data=m19.data,
    features=m8.data,
    window_size=100,
    feature_clip=5,
    flatten=True,
    window_along_col='instrument'
)

m6 = M.dl_model_train.v1(
    input_model=m5.data,
    training_data=m13.data,
    optimizer='SGD',
    loss='binary_crossentropy',
    metrics='accuracy',
    batch_size=2048,
    epochs=10,
    n_gpus=1,
    verbose='1:输出进度条记录'
)

m20 = M.filter.v3(
    input_data=m18.data,
    expr='date>\'2017-03-01\'',
    output_left_data=False
)

m14 = M.dl_convert_to_bin.v2(
    input_data=m20.data,
    features=m8.data,
    window_size=100,
    feature_clip=5,
    flatten=True,
    window_along_col='instrument'
)

m7 = M.dl_model_predict.v1(
    trained_model=m6.data,
    input_data=m14.data,
    batch_size=10240,
    n_gpus=2,
    verbose='2:每个epoch输出一行记录'
)

m2 = M.cached.v3(
    input_1=m7.data,
    input_2=m14.data,
    input_3=m20.data,
    run=m2_run_bigquant_run,
    post_run=m2_post_run_bigquant_run,
    input_ports='',
    params='{}',
    output_ports=''
)

m1 = M.trade.v4(
    instruments=m24.data,
    options_data=m2.data_1,
    start_date='2017-04-01',
    end_date='',
    handle_data=m1_handle_data_bigquant_run,
    prepare=m1_prepare_bigquant_run,
    initialize=m1_initialize_bigquant_run,
    before_trading_start=m1_before_trading_start_bigquant_run,
    volume_limit=0.025,
    order_price_field_buy='open',
    order_price_field_sell='close',
    capital_base=1000000,
    auto_cancel_non_tradable_orders=True,
    data_frequency='daily',
    price_type='真实价格',
    product_type='股票',
    plot_charts=True,
    backtest_only=False,
    benchmark=''
)

[2019-04-03 13:48:07.896277] INFO: bigquant: input_features.v1 开始运行..

[2019-04-03 13:48:07.904895] INFO: bigquant: 命中缓存

[2019-04-03 13:48:07.907236] INFO: bigquant: input_features.v1 运行完成[0.010965s].

[2019-04-03 13:48:07.962294] INFO: bigquant: instruments.v2 开始运行..

[2019-04-03 13:48:07.972887] INFO: bigquant: 命中缓存

[2019-04-03 13:48:07.977908] INFO: bigquant: instruments.v2 运行完成[0.015613s].

[2019-04-03 13:48:07.985861] INFO: bigquant: cached.v3 开始运行..

[2019-04-03 13:48:07.993918] INFO: bigquant: 命中缓存

[2019-04-03 13:48:07.995635] INFO: bigquant: cached.v3 运行完成[0.009774s].

[2019-04-03 13:48:07.999754] INFO: bigquant: cached.v3 开始运行..

[2019-04-03 13:48:08.004822] INFO: bigquant: 命中缓存

[2019-04-03 13:48:08.006620] INFO: bigquant: cached.v3 运行完成[0.006856s].

[2019-04-03 13:48:08.055531] INFO: bigquant: derived_feature_extractor.v3 开始运行..

[2019-04-03 13:48:08.063817] INFO: bigquant: 命中缓存

[2019-04-03 13:48:08.066049] INFO: bigquant: derived_feature_extractor.v3 运行完成[0.010525s].

[2019-04-03 13:48:08.073145] INFO: bigquant: join.v3 开始运行..

[2019-04-03 13:48:08.080141] INFO: bigquant: 命中缓存

[2019-04-03 13:48:08.082044] INFO: bigquant: join.v3 运行完成[0.008901s].

[2019-04-03 13:48:08.086399] INFO: bigquant: dropnan.v1 开始运行..

[2019-04-03 13:48:08.093557] INFO: bigquant: 命中缓存

[2019-04-03 13:48:08.095610] INFO: bigquant: dropnan.v1 运行完成[0.009205s].

[2019-04-03 13:48:08.169069] INFO: bigquant: filter.v3 开始运行..

[2019-04-03 13:48:08.180694] INFO: bigquant: 命中缓存

[2019-04-03 13:48:08.183099] INFO: bigquant: filter.v3 运行完成[0.014029s].

[2019-04-03 13:48:08.192979] INFO: bigquant: dl_convert_to_bin.v2 开始运行..

[2019-04-03 13:48:08.250210] ERROR: bigquant: module name: dl_convert_to_bin, module version: v2, trackeback: Traceback (most recent call last):
KeyError: 'instrument'

---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
<ipython-input-6-eeb54a6a3662> in <module>()
     97     feature_clip=5,
     98     flatten=True,
---> 99     window_along_col='instrument'
    100 )

KeyError: 'instrument'