train_catastrophe_model_human.py 文件源码

def load_features(self, episode_paths):
        features = {}
        labels = None
        for episode_path in episode_paths:
            episode_features, episode_labels = self.load_features_episode(episode_path)
            for key, value in episode_features.items():
                if key not in features:
                    features[key] = value
                else:
                    features[key] = np.concatenate([features[key], value], axis=0)
            if labels is None:
                labels = episode_labels
            else:
                labels = np.concatenate([labels, episode_labels], axis=0)
            print(episode_path)
        return features, labels