File size: 791 Bytes
3dbfcc9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from abc import abstractmethod
from transformers import PretrainedConfig
class MeasurementPredictorConfig(PretrainedConfig):
def __init__(
self,
sensor_token=" omit",
sensor_token_id=None, # 35991
n_sensors=3,
use_aggregated=True,
sensors_weight = 0.7,
aggregate_weight=0.3,
**kwargs
):
self.sensor_token = sensor_token
self.sensor_token_id = sensor_token_id
self.n_sensors = n_sensors
self.use_aggregated = use_aggregated
self.sensors_weight = sensors_weight
self.aggregate_weight = aggregate_weight
super().__init__(**kwargs)
self.emb_dim = self.get_emb_dim()
@abstractmethod
def get_emb_dim(self):
raise NotImplementedError |