Create ai_model.py

KOSASIH · Jul 20, 2024 · 5fcdba7 · 5fcdba7
1 parent 7f68129
commit 5fcdba7
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/models/ai_model.py b/models/ai_model.py
@@ -0,0 +1,15 @@
+# ai_model.py
+import tensorflow as tf
+
+# Define the AI model
+model = tf.keras.models.Sequential([
+  tf.keras.layers.Dense(64, activation="relu", input_shape=(1,)),
+  tf.keras.layers.Dense(64, activation="relu"),
+  tf.keras.layers.Dense(1)
+])
+
+# Compile the model
+model.compile(optimizer="adam", loss="mean_squared_error")
+
+# Train the model
+model.fit(X_train, y_train, epochs=10)