Create orderIA.py

orderIA.py

@@ -0,0 +1,42 @@
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, Trainer, TrainingArguments
+import pandas as pd
+from datasets import Dataset
+
+# Cargar el dataset
+data = pd.read_csv('orders.csv')
+dataset = Dataset.from_pandas(data)
+
+# Tokenizar los datos
+tokenizer = AutoTokenizer.from_pretrained("bert-base-multilingual-cased")
+
+def preprocess_function(examples):
+    return tokenizer(examples['order'], truncation=True, padding=True)
+
+tokenized_dataset = dataset.map(preprocess_function, batched=True)
+
+# Configurar el modelo
+model = AutoModelForSequenceClassification.from_pretrained("bert-base-multilingual-cased", num_labels=3)
+
+# Configurar el entrenador
+training_args = TrainingArguments(
+    output_dir='./results',
+    evaluation_strategy="epoch",
+    per_device_train_batch_size=16,
+    per_device_eval_batch_size=16,
+    num_train_epochs=3,
+    weight_decay=0.01,
+)
+
+trainer = Trainer(
+    model=model,
+    args=training_args,
+    train_dataset=tokenized_dataset,
+    eval_dataset=tokenized_dataset,
+)
+
+# Entrenar el modelo
+trainer.train()
+
+# Guardar el modelo
+model.save_pretrained("SMARTORDERIA")
+tokenizer.save_pretrained("SMARTORDERIA")