Modified gem5 with new, machine learning based cache management policy