# encoding=utf-8
"""
@author: xiao nian
@contact: xiaonian030@163.com
@time: 2021-12-14 15:15
"""

from utils.train import train_model
from utils.data_process import data as data_process


# 处理数据
print('开始处理数据')
train_x, train_y, valid_x, valid_y, test_x, test_y = data_process()
print(f"训练数据样本: {len(train_x)} 分类标签： {len(train_y)}")
print(f"评估数据样本: {len(valid_x)} 分类标签： {len(valid_y)}")
print(f"测试数据样本: {len(test_x)} 分类标签： {len(test_y)}")

# 开始训练
print('开始训练')
train_model(
    train_x,
    train_y,
    valid_x,
    valid_y,
    test_x,
    test_y)