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 = []
