@@ -47,7 +47,6 @@ def __init__(
47
47
add_image_embedding_channel : bool = False ,
48
48
include_gsp_yield_history : bool = True ,
49
49
include_sun : bool = True ,
50
- include_gsp : bool = True ,
51
50
embedding_dim : Optional [int ] = 16 ,
52
51
forecast_minutes : int = 30 ,
53
52
history_minutes : int = 60 ,
@@ -108,7 +107,6 @@ def __init__(
108
107
interval_minutes: The interval between each sample of the target data
109
108
wind_encoder: Encoder for wind data
110
109
wind_history_minutes: Length of recent wind data used as input.
111
- include_gsp: Whether to include GSP data in the model
112
110
pv_interval_minutes: The interval between each sample of the PV data
113
111
sat_interval_minutes: The interval between each sample of the satellite data
114
112
sensor_interval_minutes: The interval between each sample of the sensor data
@@ -119,7 +117,6 @@ def __init__(
119
117
self .include_nwp = nwp_encoders_dict is not None and len (nwp_encoders_dict ) != 0
120
118
self .include_pv = pv_encoder is not None
121
119
self .include_sun = include_sun
122
- self .include_gsp = include_gsp
123
120
self .include_wind = wind_encoder is not None
124
121
self .include_sensor = sensor_encoder is not None
125
122
self .embedding_dim = embedding_dim
@@ -210,7 +207,7 @@ def __init__(
210
207
if wind_history_minutes is None :
211
208
wind_history_minutes = history_minutes
212
209
213
- self .wind_encoder = wind_encoder (sequence_length = self . history_len_30 )
210
+ self .wind_encoder = wind_encoder (sequence_length = wind_history_minutes // interval_minutes )
214
211
215
212
# Update num features
216
213
fusion_input_features += self .wind_encoder .out_features
@@ -219,7 +216,7 @@ def __init__(
219
216
if sensor_history_minutes is None :
220
217
sensor_history_minutes = history_minutes
221
218
222
- self .sensor_encoder = sensor_encoder (sequence_length = self . history_len_30 )
219
+ self .sensor_encoder = sensor_encoder (sequence_length = sensor_history_minutes // sensor_interval_minutes )
223
220
224
221
# Update num features
225
222
fusion_input_features += self .sensor_encoder .out_features
0 commit comments