def __init__(self, data): X_train, X_test, \ self.y_name_train, self.y_name_test, \ self.y_topic_train, self.y_topic_test = get_train_test_split(data, TRAIN_TEST_SPLIT) self.max_features = 25000 # how many unique words to use (i.e num rows in embedding vector) self.maxlen = 100 self.epochs = EPOCHS self.embed_size = EMBEDDING_SIZE embedding = Embedding(X_train, X_test, self.max_features, self.maxlen) self.embedding_matrix = embedding.generate_embedding_matrix() self.X_train, self.X_test = embedding.t_X_train, embedding.t_X_test