tensorflow 基础（1）预测广告投出和利润

#!/usr/bin/env python
# coding: utf-8

# In[7]:


import tensorflow as tf
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
get_ipython().run_line_magic('matplotlib', 'inline')


# In[8]:


data=pd.read_csv('./Desktop/tfData/Advertising.csv')


# In[9]:


data


# In[10]:


plt.scatter(data.radio,data.sales)


# In[11]:


plt.scatter(data.TV,data.sales)


# In[12]:


plt.scatter(data.newspaper,data.sales)


# In[13]:


x=data.iloc[:,1:-1]
y=data.iloc[:,-1]


# In[14]:


x,y


# In[18]:


model=tf.keras.Sequential([tf.keras.layers.Dense(10,input_shape=(3,),activation='relu'),
                          tf.keras.layers.Dense(1)])


# In[19]:


model.summary()


# In[20]:


model.compile(optimizer='adam',loss='mse')


# In[22]:


model.fit(x,y,epochs=200)


# In[24]:


model.predict(data.iloc[:10,1:-1])


# In[26]:


data.iloc[:10,-1]


# In[ ]:





# In[ ]: