Esempio n. 1
0
    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