Basic training data extractor for stacking

Signed-off-by: Vidhoon Viswanathan <vidhu2366@gmail.com>