基本回归：预测燃油效率

“”"
Basic regression: Predict fuel efficiency
汽车燃油效率
“”"

import pathlib

import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns

import tensorflow as tf

from tensorflow import keras
from tensorflow.keras import layers

print(tf.version)

dataset_path = keras.utils.get_file(‘auto-mpg.data’,
‘http://archive.ics.uci.edu/ml/machine-learning-databases/auto-mpg/auto-mpg.data’)
print(dataset_path)
#导入数据集
column_names = [‘MPG’,‘Cylinders’,‘Displacement’,‘Horsepower’,‘Weight’,
‘Acceleration’, ‘Model Year’, ‘Origin’]
raw_dataset = pd.read_csv(dataset_path,
names=column_names,
na_values = “?”,
comment=’\t’,
sep=" ",
skipinitialspace=True)

dataset = raw_dataset.copy()
dataset.tail()
#数据清洗
#查看未知数据
print(dataset.isna().sum())
#删除未知数据
dataset = dataset.dropna()

origin = dataset.pop(‘Origin’)

dataset[‘USA’] = (origin == 1)*1.0
dataset[‘Europe’] = (origin == 2)*1.0
dataset[‘Japan’] = (origin == 3)*1.0
dataset.tail()

#拆分训练数据集和测试数据集
train_dataset = dataset.sample