Setup dotnet wrapper around Python AI project for UI
This commit is contained in:
@@ -0,0 +1,39 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user