from __future__ import absolute_import, division, print_function

# TensorFlow and tf.keras
import tensorflow as tf
from tensorflow import keras

# Helper libraries
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

from utils import *

tf.enable_eager_execution()

df = pd.read_pickle('data_final.p')
# print(df.head())

# dataset = dataframe_to_dataset_biomes(df)
dataset = dataframe_to_dataset_temp_precip(df)

i = 0
for feature, target in dataset:
    i += 1
    if i > 10:
        break
    print('{} => {}'.format(feature, target))

print(tf.__version__)