π¬Tensor Flow
// Simple Shopzzy AI script utilizing TensorFlow technology
import tensorflow as tf
import numpy as np
# Sample data
train_data = [
("Nike Soccer Ball", "sports"),
("The North Face Winter Jacket", "outdoor"),
("Samsung Galaxy Smartphone", "electronics"),
("Lavazza Coffee Beans", "food"),
("Asics Running Shoes", "sports"),
("GoPro Camera", "electronics"),
("Deuter Trekking Backpack", "outdoor"),
("Garmin Sports Watch", "sports")
]
# Prepare training data
train_x = [x[0] for x in train_data]
train_y = [x[1] for x in train_data]
# Tokenize input data
tokenizer = tf.keras.preprocessing.text.Tokenizer()
tokenizer.fit_on_texts(train_x)
train_sequences = tokenizer.texts_to_sequences(train_x)
# Pad input data
train_padded = tf.keras.preprocessing.sequence.pad_sequences(train_sequences)
# Convert output data to one-hot encoding
train_labels = np.zeros((len(train_y), len(set(train_y))))
for i, label in enumerate(train_y):
train_labels[i, train_y.index(label)] = 1
# Neural network model
model = tf.keras.Sequential([
tf.keras.layers.Embedding(len(tokenizer.word_index)+1, 64),
tf.keras.layers.Bidirectional(tf.keras.layers.LSTM(64)),
tf.keras.layers.Dense(64, activation='relu'),
tf.keras.layers.Dense(len(set(train_y)), activation='softmax')
])
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
# Train the model
model.fit(train_padded, train_labels, epochs=10)
# Sample product description
product_description = "Adidas Soccer Ball"
# Tokenize and pad product description
product_sequence = tokenizer.texts_to_sequences([product_description])
product_padded = tf.keras.preprocessing.sequence.pad_sequences(product_sequence)
# Classify product
class_index = np.argmax(model.predict(product_padded), axis=-1)[0]
predicted_class = list(set(train_y))[class_index]
# Print result
print(f"Product description: {product_description}")
print(f"Predicted category: {predicted_class}")Last updated