Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import functools
- import tensorflow as tf
- import numpy as np
- import os
- train_dataset_file = open("./StarterData/1504218.csv")
- CSV_COLUMNS = ['Date', 'Time', 'System Setting', 'System Mode', "Calendar Event", "Program Mode", "Cool Set Temp (F)", "Heat Set Temp (F)", "Current Temp (F)",
- "Current Humidity (%RH)", "Outdoor Temp (F)", "Wind Speed (km/h)", "Cool Stage 1 (sec)", "Heat Stage 1 (sec)", "Fan (sec)", "DM Offset", "Thermostat Temperature", "Thermostat Humidity (%RH)"]
- feature_names = CSV_COLUMNS[:-1]
- label_column = 'Current Temp (F)'
- np.set_printoptions(precision=3, suppress=True)
- def get_dataset(file_path, **kwargs):
- dataset = tf.data.experimental.make_csv_dataset(
- file_path,
- batch_size=32,
- label_name=label_column,
- na_value="?",
- num_epochs=1,
- ignore_errors=True,
- **kwargs)
- return dataset
- temp_dataset = get_dataset(train_dataset_file, column_names = CSV_COLUMNS)
- def show_batch(dataset):
- print('data')
- for batch, label in dataset.take(1):
- for key, value in batch.items():
- print("{:20s} {}".format(key, value.numpy()))
- show_batch(temp_dataset)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement