Ejemplo n.º 1
0
def make_pipeline():
   
    value = Fundamentals.ebit.latest / Fundamentals.enterprise_value.latest
    sentiment_score = SimpleMovingAverage(
        inputs=[stocktwits.total_scanned_messages],
        window_length=21,
    )
    
    ty = Fundamentals.total_yield.latest 
    wps = Fundamentals.working_capital_per_share.latest 
    gro = Fundamentals.growth_score.latest
    dollar_volume = AverageDollarVolume(window_length=63)
    
    universe = QTradableStocksUS()
    
    dollar_winsorize = dollar_volume.winsorize(min_percentile=0.2, max_percentile=0.8)
    value_winsorized = value.winsorize(min_percentile=0.05, max_percentile=0.95)
    sentiment_score_winsorized = sentiment_score.winsorize(min_percentile=0.05, max_percentile=0.95)
    eps_wisorized = wps.winsorize(min_percentile=0.05, max_percentile=0.95)
    ty_winsorized = ty.winsorize(min_percentile=0.05, max_percentile=0.95)
    gro_winsorize = gro.winsorize(min_percentile=0.05, max_percentile=0.95)

   
    combined_factor = (
        value_winsorized.zscore() *0.15
        + gro_winsorize.zscore() *0.25 
        + sentiment_score_winsorized.zscore()* 0.15
        + eps_wisorized.zscore() *0.15 
        + ty_winsorized.zscore() *0.15
        + dollar_winsorize.zscore() *0.15
    )

    longs = combined_factor.top(TOTAL_POSITIONS//2, mask=universe)
    shorts = combined_factor.bottom(TOTAL_POSITIONS//2, mask=universe)

    long_short_screen = (longs | shorts)

    pipe = Pipeline(
        columns={
            'longs': longs,
            'shorts': shorts,
            'combined_factor': combined_factor
        },
        screen=long_short_screen
    )


    return pipe
def make_pipeline():
    """
    A function that creates and returns our pipeline.

    We break this piece of logic out into its own function to make it easier to
    test and modify in isolation. In particular, this function can be
    copy/pasted into research and run by itself.

    Returns
    -------
    pipe : Pipeline
        Represents computation we would like to perform on the assets that make
        it through the pipeline screen.
    """
    # The factors we create here are based on fundamentals data and a moving
    # average of sentiment data
    value = Fundamentals.ebit.latest / Fundamentals.enterprise_value.latest
    quality = Fundamentals.roe.latest
    sentiment_score = SimpleMovingAverage(
        inputs=[stocktwits.bull_minus_bear],
        window_length=3,
    )
    
    sentiment_week = SimpleMovingAverage(
        inputs=[stocktwits.bull_minus_bear],
        window_length=7,
    )
    
    sentiment_2weeks = SimpleMovingAverage(
        inputs=[stocktwits.bull_minus_bear],
        window_length=15,
    )
    
    dollar_volume = AverageDollarVolume(window_length=15)

    universe = QTradableStocksUS()
    
    # We winsorize our factor values in order to lessen the impact of outliers
    # For more information on winsorization, please see
    # https://en.wikipedia.org/wiki/Winsorizing
    value_winsorized = value.winsorize(min_percentile=0.05, max_percentile=0.95)
    quality_winsorized = quality.winsorize(min_percentile=0.05, max_percentile=0.95)
    sentiment_score_winsorized = sentiment_score.winsorize(min_percentile=0.05,                                                                             max_percentile=0.95)
    sentiment_week_winsorized = sentiment_week.winsorize(min_percentile=0.05,                                                                             max_percentile=0.95)
    sentiment_2weeks_winsorized = sentiment_2weeks.winsorize(min_percentile=0.05,                                                                             max_percentile=0.95)
    dollar_volume_winsorized = dollar_volume.winsorize(min_percentile=0.05,                                                                             max_percentile=0.95)

    # Here we combine our winsorized factors, z-scoring them to equalize their influence
    combined_factor = (
        value_winsorized.zscore() + 
        quality_winsorized.zscore() + 
        dollar_volume_winsorized.zscore() +
        sentiment_score_winsorized.zscore()/sentiment_2weeks_winsorized.zscore() +
        sentiment_week_winsorized.zscore()/sentiment_2weeks_winsorized.zscore()
    )

    # Build Filters representing the top and bottom baskets of stocks by our
    # combined ranking system. We'll use these as our tradeable universe each
    # day.
    longs = combined_factor.top(TOTAL_POSITIONS//2, mask=universe)
    shorts = combined_factor.bottom(TOTAL_POSITIONS//2, mask=universe)

    # The final output of our pipeline should only include
    # the top/bottom 300 stocks by our criteria
    long_short_screen = (longs | shorts)

    # Create pipeline
    pipe = Pipeline(
        columns={
            'longs': longs,
            'shorts': shorts,
            'combined_factor': combined_factor
        },
        screen=long_short_screen
    )
    return pipe