Skip to content

Commit

Permalink
New Fundamental Data (#7490)
Browse files Browse the repository at this point in the history
* New Fundamental Data

* Minor CIK lookup fix

* Handle live mode & delete unexisting properties

* Minor coarse fundamental adjustment

* Add fundamental history support

* Fix unit tests

* Performance improvements

* Fixes

* Minor regression algorithm fix

* Improvements. Add FundamentalUniverseSelectionModel

* Change default values

* Fix unit test

* Minor tweaks

* Fix unit test

* Minor error handling improvement

* Fix rebase
  • Loading branch information
Martin-Molinero authored Oct 25, 2023
1 parent 4411d4a commit f8b258d
Show file tree
Hide file tree
Showing 906 changed files with 111,690 additions and 62,989 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -169,7 +169,7 @@ public override void OnEndOfAlgorithm()
/// <summary>
/// Data Points count of all timeslices of algorithm
/// </summary>
public long DataPoints => 5797;
public long DataPoints => 5798;

/// <summary>
/// Data Points count of the algorithm history
Expand Down
8 changes: 4 additions & 4 deletions Algorithm.CSharp/Alphas/GreenblattMagicFormulaAlpha.cs
Original file line number Diff line number Diff line change
Expand Up @@ -195,11 +195,11 @@ private class GreenBlattMagicFormulaUniverseSelectionModel : FundamentalUniverse
private const int _numberOfSymbolsFine = 20;
private const int _numberOfSymbolsInPortfolio = 10;
private int _lastMonth = -1;
private Dictionary<Symbol, decimal> _dollarVolumeBySymbol;
private Dictionary<Symbol, double> _dollarVolumeBySymbol;

public GreenBlattMagicFormulaUniverseSelectionModel() : base(true)
{
_dollarVolumeBySymbol = new Dictionary<Symbol, decimal>();
_dollarVolumeBySymbol = new ();
}

/// <summary>
Expand Down Expand Up @@ -245,7 +245,7 @@ from x in fine
where x.CompanyReference.CountryId == "USA"
where x.CompanyReference.PrimaryExchangeID == "NYS" || x.CompanyReference.PrimaryExchangeID == "NAS"
where (algorithm.Time - x.SecurityReference.IPODate).TotalDays > 180
where x.EarningReports.BasicAverageShares.ThreeMonths * x.EarningReports.BasicEPS.TwelveMonths * x.ValuationRatios.PERatio > 5e8m
where x.EarningReports.BasicAverageShares.ThreeMonths * x.EarningReports.BasicEPS.TwelveMonths * x.ValuationRatios.PERatio > 5e8
select x;

double count = filteredFine.Count();
Expand Down Expand Up @@ -287,4 +287,4 @@ select item
}
}
}
}
}
4 changes: 2 additions & 2 deletions Algorithm.CSharp/CoarseFineFundamentalRegressionAlgorithm.cs
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,7 @@ public void OnData(TradeBars data)
// we want 50% allocation in each security in our universe
foreach (var security in _changes.AddedSecurities)
{
if (security.Fundamentals.EarningRatios.EquityPerShareGrowth.OneYear > 0.25m)
if (security.Fundamentals.EarningRatios.EquityPerShareGrowth.OneYear > 0.25)
{
SetHoldings(security.Symbol, 0.5m);
Debug("Purchased Stock: " + security.Symbol.Value);
Expand Down Expand Up @@ -159,7 +159,7 @@ public override void OnSecuritiesChanged(SecurityChanges changes)
/// <summary>
/// Data Points count of all timeslices of algorithm
/// </summary>
public long DataPoints => 7239;
public long DataPoints => 7244;

/// <summary>
/// Data Points count of the algorithm history
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -157,7 +157,7 @@ public override void OnEndOfAlgorithm()
/// <summary>
/// Data Points count of all timeslices of algorithm
/// </summary>
public long DataPoints => 998462;
public long DataPoints => 998464;

/// <summary>
/// Data Points count of the algorithm history
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ public override void OnData(Slice data)
/// <summary>
/// Data Points count of all timeslices of algorithm
/// </summary>
public long DataPoints => 40;
public long DataPoints => 41;

/// <summary>
/// Data Points count of the algorithm history
Expand Down
201 changes: 201 additions & 0 deletions Algorithm.CSharp/FundamentalRegressionAlgorithm.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,201 @@
/*
* QUANTCONNECT.COM - Democratizing Finance, Empowering Individuals.
* Lean Algorithmic Trading Engine v2.0. Copyright 2014 QuantConnect Corporation.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

using System;
using System.Linq;
using QuantConnect.Interfaces;
using QuantConnect.Data.Market;
using System.Collections.Generic;
using QuantConnect.Data.Fundamental;
using QuantConnect.Data.UniverseSelection;
using QuantConnect.Data;

namespace QuantConnect.Algorithm.CSharp
{
/// <summary>
/// Demonstration of how to define a universe using the fundamental data
/// </summary>
public class FundamentalRegressionAlgorithm : QCAlgorithm, IRegressionAlgorithmDefinition
{
private const int NumberOfSymbolsFundamental = 2;

private SecurityChanges _changes = SecurityChanges.None;

public override void Initialize()
{
UniverseSettings.Resolution = Resolution.Daily;

SetStartDate(2014, 03, 25);
SetEndDate(2014, 04, 07);

AddEquity("SPY");
AddEquity("AAPL");

// Request fundamental data for symbols at current algorithm time
var ibm = QuantConnect.Symbol.Create("IBM", SecurityType.Equity, Market.USA);
var ibmFundamental = Fundamentals(ibm);
if (Time != StartDate || Time != ibmFundamental.EndTime)
{
throw new Exception($"Unexpected {nameof(Fundamental)} time {ibmFundamental.EndTime}");
}
if (ibmFundamental.Price == 0)
{
throw new Exception($"Unexpected {nameof(Fundamental)} IBM price!");
}

var nb = QuantConnect.Symbol.Create("NB", SecurityType.Equity, Market.USA);
var fundamentals = Fundamentals(new List<Symbol>{ nb, ibm }).ToList();
if (fundamentals.Count != 2)
{
throw new Exception($"Unexpected {nameof(Fundamental)} count {fundamentals.Count}! Expected 2");
}

// Request historical fundamental data for symbols
var history = History<Fundamental>(Securities.Keys, new TimeSpan(1, 0, 0, 0)).ToList();
if(history.Count != 1)
{
throw new Exception($"Unexpected {nameof(Fundamental)} history count {history.Count}! Expected 1");
}

if (history[0].Values.Count != 2)
{
throw new Exception($"Unexpected {nameof(Fundamental)} data count {history[0].Values.Count}, expected 2!");
}

foreach (var ticker in new[] {"AAPL", "SPY"})
{
if (!history[0].TryGetValue(ticker, out var fundamental) || fundamental.Price == 0)
{
throw new Exception($"Unexpected {ticker} fundamental data");
}
}

// Request historical fundamental data for all symbols
var history2 = History<Fundamentals>(new TimeSpan(1, 0, 0, 0)).ToList();
if (history2.Count != 1)
{
throw new Exception($"Unexpected {nameof(Fundamentals)} history count {history.Count}! Expected 1");
}
if (history2[0].Single().Value.Data.Count < 7000)
{
throw new Exception($"Unexpected {nameof(Fundamentals)} data count {history.Count}! Expected > 7000");
}
if (history2[0].Single().Value.Data.Any(x => x.GetType() != typeof(Fundamental)))
{
throw new Exception($"Unexpected {nameof(Fundamentals)} data type!");
}

AddUniverse(FundamentalSelectionFunction);
}

// sort the data by daily dollar volume and take the top 'NumberOfSymbolsCoarse'
public IEnumerable<Symbol> FundamentalSelectionFunction(IEnumerable<Fundamental> fundamental)
{
// select only symbols with fundamental data and sort descending by daily dollar volume
var sortedByDollarVolume = fundamental
.Where(x => x.Price > 1)
.OrderByDescending(x => x.DollarVolume);

// sort descending by P/E ratio
var sortedByPeRatio = sortedByDollarVolume.OrderByDescending(x => x.ValuationRatios.PERatio);

// take the top entries from our sorted collection
var topFine = sortedByPeRatio.Take(NumberOfSymbolsFundamental);

// we need to return only the symbol objects
return topFine.Select(x => x.Symbol);
}

public override void OnData(Slice slice)
{
// if we have no changes, do nothing
if (_changes == SecurityChanges.None) return;

// liquidate removed securities
foreach (var security in _changes.RemovedSecurities)
{
if (security.Invested)
{
Liquidate(security.Symbol);
}
}

// we want allocation in each security in our universe
foreach (var security in _changes.AddedSecurities)
{
SetHoldings(security.Symbol, 0.02m);
}

_changes = SecurityChanges.None;
}

// this event fires whenever we have changes to our universe
public override void OnSecuritiesChanged(SecurityChanges changes)
{
_changes = changes;
}

/// <summary>
/// This is used by the regression test system to indicate if the open source Lean repository has the required data to run this algorithm.
/// </summary>
public bool CanRunLocally { get; } = true;

/// <summary>
/// This is used by the regression test system to indicate which languages this algorithm is written in.
/// </summary>
public Language[] Languages { get; } = { Language.CSharp, Language.Python };

/// <summary>
/// Data Points count of all timeslices of algorithm
/// </summary>
public long DataPoints => 85867;

/// <summary>
/// Data Points count of the algorithm history
/// </summary>
public virtual int AlgorithmHistoryDataPoints => 3;

/// <summary>
/// This is used by the regression test system to indicate what the expected statistics are from running the algorithm
/// </summary>
public Dictionary<string, string> ExpectedStatistics => new Dictionary<string, string>
{
{"Total Trades", "2"},
{"Average Win", "0%"},
{"Average Loss", "0%"},
{"Compounding Annual Return", "-0.223%"},
{"Drawdown", "0.100%"},
{"Expectancy", "0"},
{"Net Profit", "-0.009%"},
{"Sharpe Ratio", "-6.313"},
{"Probabilistic Sharpe Ratio", "12.055%"},
{"Loss Rate", "0%"},
{"Win Rate", "0%"},
{"Profit-Loss Ratio", "0"},
{"Alpha", "-0.019"},
{"Beta", "0.027"},
{"Annual Standard Deviation", "0.004"},
{"Annual Variance", "0"},
{"Information Ratio", "1.749"},
{"Tracking Error", "0.095"},
{"Treynor Ratio", "-0.876"},
{"Total Fees", "$2.00"},
{"Estimated Strategy Capacity", "$2200000000.00"},
{"Lowest Capacity Asset", "IBM R735QTJ8XC9X"},
{"Portfolio Turnover", "0.28%"},
{"OrderListHash", "34bb9933f9d242713c0ec14c4ee586b6"}
};
}
}
101 changes: 101 additions & 0 deletions Algorithm.CSharp/FundamentalUniverseSelectionRegressionAlgorithm.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,101 @@
/*
* QUANTCONNECT.COM - Democratizing Finance, Empowering Individuals.
* Lean Algorithmic Trading Engine v2.0. Copyright 2014 QuantConnect Corporation.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

using System.Linq;
using QuantConnect.Data;
using System.Collections.Generic;
using QuantConnect.Data.Fundamental;
using QuantConnect.Data.UniverseSelection;
using QuantConnect.Algorithm.Framework.Selection;

namespace QuantConnect.Algorithm.CSharp
{
/// <summary>
/// Demonstration of how to define a universe using the fundamental data
/// </summary>
public class FundamentalUniverseSelectionRegressionAlgorithm : FundamentalRegressionAlgorithm
{
private const int NumberOfSymbolsFundamental = 2;

private SecurityChanges _changes = SecurityChanges.None;

public override void Initialize()
{
UniverseSettings.Resolution = Resolution.Daily;

SetStartDate(2014, 03, 25);
SetEndDate(2014, 04, 07);

AddEquity("SPY");
AddEquity("AAPL");

SetUniverseSelection(new FundamentalUniverseSelectionModelTest());
}

private class FundamentalUniverseSelectionModelTest : FundamentalUniverseSelectionModel
{
public override IEnumerable<Symbol> Select(QCAlgorithm algorithm, IEnumerable<Fundamental> fundamental)
{
// select only symbols with fundamental data and sort descending by daily dollar volume
var sortedByDollarVolume = fundamental
.Where(x => x.Price > 1)
.OrderByDescending(x => x.DollarVolume);

// sort descending by P/E ratio
var sortedByPeRatio = sortedByDollarVolume.OrderByDescending(x => x.ValuationRatios.PERatio);

// take the top entries from our sorted collection
var topFine = sortedByPeRatio.Take(NumberOfSymbolsFundamental);

// we need to return only the symbol objects
return topFine.Select(x => x.Symbol);
}
}

public override void OnData(Slice slice)
{
// if we have no changes, do nothing
if (_changes == SecurityChanges.None) return;

// liquidate removed securities
foreach (var security in _changes.RemovedSecurities)
{
if (security.Invested)
{
Liquidate(security.Symbol);
}
}

// we want allocation in each security in our universe
foreach (var security in _changes.AddedSecurities)
{
SetHoldings(security.Symbol, 0.02m);
}

_changes = SecurityChanges.None;
}

// this event fires whenever we have changes to our universe
public override void OnSecuritiesChanged(SecurityChanges changes)
{
_changes = changes;
}

/// <summary>
/// Data Points count of the algorithm history
/// </summary>
public override int AlgorithmHistoryDataPoints => 0;
}
}
Loading

0 comments on commit f8b258d

Please sign in to comment.