Пример #1
0
 def __init__(self, **kwargs):
     self.transmission = basicTransmission.Transmssion()
     self.HumanInput = utils.HumanInput()
     self.hs_counter = {
         'left': 0,
         'right': 0
     }
Пример #2
0
 def __init__(self, speed_limits=settings['speed_limits'], **kwargs):
     self.speed_limits = speed_limits
     self.track_divison = track_divison = np.linspace(
         0.,
         settings['track_length'],
         len(self.speed_limits) + 1
     )
     self.is_starting = True
     self.transmission = basicTransmission.Transmssion()
Пример #3
0
    def __init__(self, model_path=None, dump_after=True, **kwargs):
        self.transmission = basicTransmission.Transmssion()

        if model_path is None:
            self.state_keys = settings['state_keys']
            data = parse_dataset()
            states, actions = extract_parameters(data)
            # self.actions = actions
            # Reducinf actions set
            Y, self.actions = unify_actions(actions)

            X = np.array(states)
            self.X_max = X.max(axis=0)
            self.X_min = X.min(axis=0)
            X_normed = (X - self.X_min) / (self.X_max - self.X_min)
            self.knn.fit(X_normed, Y)

            if dump_after:
                self.dump()

        else:
            self.load(model_path)
 def __init__(self, model_path, **kwargs):
     self.transmission = basicTransmission.Transmssion()
     self.load(model_path)