MOOC网深度学习及其应用1——神经网络基础

最新推荐文章于 2024-02-10 02:30:58 发布

123梦野

最新推荐文章于 2024-02-10 02:30:58 发布

阅读量155

点赞数

分类专栏： tensorflow

本文链接：https://blog.csdn.net/qq_42832272/article/details/118337190

版权

tensorflow 专栏收录该内容

29 篇文章 5 订阅

订阅专栏

神经网络基础

神经网络简介

from random import shuffle

import pandas as pd
import tensorflow as tf
import numpy as np


df = pd.read_csv("./data/select-data.csv")
df_test = pd.read_csv("./data/scalar-test.csv")

#构建训练向量
train=[]
target=[]
#将分类条件保存到train数组中，把最终类别保存到target中
for i in range(0,len(df["EstimatedSalary"])):
    mid = []
    mid.append(df["Geography"][i])
    mid.append(df["Gender"][i])
    mid.append(df["EB"][i])
    mid.append(df["Age"][i])
    mid.append(df["EstimatedSalart"][i])
    mid.append(df["NumOfProducts"][i])
    mid.append(df["CreditScore"][i])
    mid.append(df["Tenure"][i])
    mid.append(df["HasCrCard"][i])
    mid.append(df["IsActiveMember"][i])
    target.append(df["Exited"][i])
    train.append(mid)
#转换为np.array格式
train = np.array(train)
target= np.array(target)
#构建测试向量
test=[]
test_target=[]
#将分类条件保存到train数组中，把最终类别保存到target中
for i in range(0,len(df_test["EstimatedSalary"])):
    mid = []
    mid.append(df_test["Geography"][i])
    mid.append(df_test["Gender"][i])
    mid.append(df_test["EB"][i])
    mid.append(df_test["Age"][i])
    mid.append(df_test["EstimatedSalart"][i])
    mid.append(df_test["NumOfProducts"][i])
    mid.append(df_test["CreditScore"][i])
    mid.append(df_test["Tenure"][i])
    mid.append(df_test["HasCrCard"][i])
    mid.append(df_test["IsActiveMember"][i])
    test_target.append(df_test["Exited"][i])
    test.append(mid)
#转换为np.array格式
test = np.array(test)
test_target= np.array(test_target)
#随机打乱训练集和标签
train,target = shuffle(train,target)
target = target.reshape(-1,1)
test_target = test_target.reshape(-1,1)
#One-Hot编码