def __init__(self, df, signal_indices): # replace infinite values SignalClassifier.__init__(self, df, signal_indices) m_dir = os.path.dirname(__file__) base_dir = os.path.join(m_dir, '../models/head_sentiment/') model_dir = os.path.join(base_dir, 'head_turns') self.model = joblib.load('%s/head_turns.pkl' % model_dir) self.active_features = \ pd.read_csv('%s/active_features.csv' % model_dir)['columns'].tolist()
def __init__(self, df, lane_indices, head_indices, head_events_hash, head_events_sentiment): SignalClassifier.__init__(self, df, lane_indices) self.head_indices = head_indices self.head_events_hash = head_events_hash self.head_events_sentiment = head_events_sentiment self.head_turns_to_keep = []