10.28 多层感知器

loc函数：通过行索引 “Index” 中的具体值来取行数据（如取"Index"为"A"的行）

iloc函数：通过行号来取行数据（如取第二行的数据）

Pandas中loc和iloc函数用法详解（源码+实例）

import tensorflow as tf
from tensorflow.keras import layers
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

data = pd.read_csv(r'c:\Users\Administrator\Desktop\Advertising.csv')
print(data)

# fig = plt.figure()
# plt.scatter(data.TV, data.sales)
# plt.show()

x = data.iloc[:, 1:-1]
y = data.iloc[:, -1]
#input_shape 以元组的形式把输入数据的形状告诉它，本题输入数据为 TV,radio，newspaper
model = tf.keras.Sequential()
model.add(tf.keras.layers.Dense(10, input_shape=(3,), activation="relu"))
#输出层
model.add(tf.keras.layers.Dense(1))

model.compile(optimizer='adam', loss= 'mse')
model.fit(x, y,epochs=100)

test=data.iloc[:10,1:-1]
prediction=model.predict(test)
print(prediction)