Overall Statistics
Total Orders
68
Average Win
2.82%
Average Loss
-1.93%
Compounding Annual Return
15.027%
Drawdown
9.400%
Expectancy
0.450
Start Equity
100000
End Equity
132314.32
Net Profit
32.314%
Sharpe Ratio
0.978
Sortino Ratio
0.411
Probabilistic Sharpe Ratio
49.954%
Loss Rate
41%
Win Rate
59%
Profit-Loss Ratio
1.47
Alpha
0.096
Beta
0.027
Annual Standard Deviation
0.102
Annual Variance
0.01
Information Ratio
-0.293
Tracking Error
0.137
Treynor Ratio
3.71
Total Fees
$9858.89
Estimated Strategy Capacity
$3500000.00
Lowest Capacity Asset
TSLA UNU3P8Y3WFAD
Portfolio Turnover
9.32%
 
# region imports
from AlgorithmImports import *
from nltk.sentiment import SentimentIntensityAnalyzer
#https://www.youtube.com/watch?v=X7XwkHsE-4Y&list=PLtqRgJ_TIq8ZtMwEYCKLC9hGdy3pO1gel&index=19
# endregion

class UpgradedFluorescentYellowAlpaca(QCAlgorithm):

    def Initialize(self):
        self.SetStartDate(2012, 11, 1)
        self.SetEndDate(2014, 11, 1)
        self.SetCash(100000)
        
        self.tsla = self.AddEquity("TSLA", Resolution.Minute).Symbol
        self.musk = self.AddData(MuskTweet, "MUSKTWTS", Resolution.Minute).Symbol
        
        self.Schedule.On(self.DateRules.EveryDay(self.tsla),
                 self.TimeRules.BeforeMarketClose(self.tsla, 15),      
                 self.ExitPositions)

    def OnData(self, data):
        if self.musk in data:
            score = data[self.musk].Value
            content = data[self.musk].Tweet
            
            if score > 0.5:
                self.SetHoldings(self.tsla, 1)
            elif score < -0.5:
                self.SetHoldings(self.tsla, -1)
                
            if abs(score) > 0.5:
                self.Log("Score: " + str(score) + ", Tweet: " + content)

    def ExitPositions(self):
        self.Liquidate()


class MuskTweet(PythonData):

    sia = SentimentIntensityAnalyzer()

    def GetSource(self, config, date, isLive):
        #source = "https://www.dropbox.com/scl/fi/2kxh30cf0j3j4vvxtsion/MuskTweetsPreProcessed.csv?rlkey=eanc82utyrlabfm8mzcms01u2&st=f0yghb7e&dl=1"
        source = "https://www.dropbox.com/s/ovnsrgg1fou1y0r/MuskTweetsPreProcessed.csv?dl=1"
        return SubscriptionDataSource(source, SubscriptionTransportMedium.RemoteFile);

    def Reader(self, config, line, date, isLive):
        if not (line.strip() and line[0].isdigit()):
            return None
        
        data = line.split(',')
        tweet = MuskTweet()
        
        try:
            tweet.Symbol = config.Symbol
            tweet.Time = datetime.strptime(data[0], '%Y-%m-%d %H:%M:%S') + timedelta(minutes=1)
            content = data[1].lower()
            
            if "tsla" in content or "tesla" in content:
                tweet.Value = self.sia.polarity_scores(content)["compound"]
            else:
                tweet.Value = 0
            
            tweet["Tweet"] = str(content)
            
        except ValueError:
            return None
        
        return tweet