derek holloway
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
from datetime import date, datetime
|
|
from datapuller import DataPuller
|
|
from database import DataBase
|
|
|
|
def main():
|
|
|
|
# Load the database object
|
|
db = DataBase()
|
|
|
|
# If we havent already pulled the stock data today
|
|
if db.GetKey("LastRun") != str(date.today()):
|
|
# Update the data store in the data folder
|
|
DataPuller.pull()
|
|
# Update the last run to today
|
|
db.SetKey("LastRun", str(date.today()))
|
|
|
|
# Lazy Load in the AI algorithms so yfinance works properly
|
|
import tensorflow as tf
|
|
import keras
|
|
from keras.layers import Dense, Flatten, Conv2D
|
|
from keras import Model
|
|
|
|
mnist = keras.datasets.mnist
|
|
|
|
(x_train, y_train), (x_test, y_test) = mnist.load_data()
|
|
x_train, x_test = x_train / 255.0, x_test / 255.0
|
|
|
|
# Add a channels dimension
|
|
x_train = x_train[..., tf.newaxis].astype("float32")
|
|
x_test = x_test[..., tf.newaxis].astype("float32")
|
|
|
|
# batch and shuffle the dataset
|
|
train_ds = tf.data.Dataset.from_tensor_slices(
|
|
(x_train, y_train)).shuffle(10000).batch(32)
|
|
|
|
test_ds = tf.data.Dataset.from_tensor_slices((x_test, y_test)).batch(32)
|
|
|
|
if __name__ == "__main__":
|
|
main() |