Use load_data in feature_forest