Team Ai
Datasetpublic

rohanjain2312/candlestick-pattern-recognition-system-data

Candlestick pattern detection dataset 640x640 renderings of 20-candle daily OHLC windows, with YOLO-format boxes for 8 candlestick patterns. Labels are rule-based, not human-verified Every box was generated by TA-Lib's CDLxxx functions. Nobody annotated these charts by hand. A model trained here learns to imitate TA-Lib's published heuristic, which does not always agree with what a discretionary trader would mark on the same chart. Treat the labels as a… See the full description on the dataset page: https://huggingface.co/datasets/rohanjain2312/candlestick-pattern-recognition-system-data.

sourceHugging Facemitupdated 22d agoView on Hugging Face
0likes70downloads
Dataset Card

Candlestick pattern detection dataset

640x640 renderings of 20-candle daily OHLC windows, with YOLO-format boxes for 8 candlestick patterns.

Labels are rule-based, not human-verified

Every box was generated by TA-Lib's CDLxxx functions. Nobody annotated these charts by hand. A model trained here learns to imitate TA-Lib's published heuristic, which does not always agree with what a discretionary trader would mark on the same chart. Treat the labels as a reproducible rule, not as truth.

Splits are chronological, with an embargo

splitperiod
trainthrough 2014-12-31
val2014-12-31 to 2018-12-31
testafter 2018-12-31

Consecutive windows overlap in 19 of 20 candles, so the first 20 windows after each boundary are discarded. Val and test images therefore share no candle with any training image. Splits are never random.

Contents

classtrainvaltest
Hammer4798475557
ShootingStar2825360570
BullishEngulfing85534571178
BearishEngulfing105836731368
MorningStar1354126198
EveningStar1106144180
Doji3232229575261
Harami2044316962350
images121519861917

train also contains auxiliary large-cap tickers (AAPL, MSFT, JPM, XOM, JNJ, KO, GE, WMT, PG, T), included solely to give the rare three-candle patterns enough examples. val and test are SPY-only, so held-out metrics describe the instrument the downstream study actually uses.

Layout

images/{train,val,test}/TICKER_YYYYMMDD.png
labels/{train,val,test}/TICKER_YYYYMMDD.txt

A filename's date is the window's last candle: the chart contains bars up to and including that date and nothing after it.

Code: https://github.com/rohanjain2312/candlestick-pattern-recognition-system