import os
from pathlib import Path

os.environ.setdefault("TF_CPP_MIN_LOG_LEVEL", "2")

import tensorflow as tf


record_path = Path("weather.tfrecord")
rows = [
    {"temperature_c": 18.5, "rainfall_mm": 3.25, "will_rain": 1},
    {"temperature_c": 21.0, "rainfall_mm": 0.0, "will_rain": 0},
    {"temperature_c": 16.75, "rainfall_mm": 8.5, "will_rain": 1},
]


def float_feature(value):
    return tf.train.Feature(float_list=tf.train.FloatList(value=[value]))


def int64_feature(value):
    return tf.train.Feature(int64_list=tf.train.Int64List(value=[value]))