Пример #1
0
def load_test_data():
    test_data = News()
    test_data.set_target_names(config.TARGET_NAMES)
    data = []
    targets = []
    with open(TEST_PATH, encoding='utf-8') as f:
        list_news = f.readlines()
        for news in list_news:
            if news and len(news) > 100:
                news = news.strip()
                data.append(news[:-2])
                targets.append(int(news[-1:]))
    test_data.set_target(targets)
    test_data.set_data(data)
    return test_data