Support cache in create_classification

It's required for U-FCN