import joblib import geopandas as gpd from shapely.geometry import Point data = joblib.load('dataset_cache/training_data_2d.joblib') X, y = data['X'], data['y'] print(f"X shape: {X.shape}, y shape: {y.shape}") gdf = gpd.read_file("train/ST_training_data_updated_1130points_new.shp") print("Total points:", len(gdf))