LSTM处理多分类问题

本文介绍了如何运用LSTM解决多分类问题,详细探讨了项目中的模型选择与实现过程。
摘要由CSDN通过智能技术生成

最近由于项目需求,尝试各种模型实现多分类。

# coding=gbk
import numpy as np
import pandas as pd
import tensorflow as tf
from sklearn.model_selection import train_test_split

data = pd.read_csv('data/sample.csv', header=0, names=['jing', 'ceng', 'y', 'DEPTH', 'AC', 'SP', 'GR', 'CAL', 'RT'],
                   encoding='utf-8', low_memory=False)[1:]
categorical_columns = ['jing', 'ceng', 'y']
for f in categorical_columns:
    data[f] = data[f].map(dict(zip(data[f].unique(), range(0, data[f].nunique()))))
# shuffle(data)
target = data['y']
del data['y']
# data:[800000,8] target:[800000,1]
# 将data转为[batch_size, sequence_length, input_dimension] shape, 最后一维对应正常NLP任务的字 embedding 维度,这里取1
# 而target转为[batch_size, output_dimension] shape, 这里直接用pd.get_dummy转换
train_input, train_outp
评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值