Working base for tensorflow without prediction of trained models

TailUFPB · Apr 2, 2024 · 9be608f · 9be608f
1 parent c43bbc0
commit 9be608f
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 4 deletions.
diff --git a/api/DataProcesser.py b/api/DataProcesser.py
@@ -84,6 +84,7 @@ def pretrained_predict(self, df, model_name):
  pipeline = pickle.load(model)
 
  texts_to_predict = df['input_column']
+ texts_to_predict = [str(text) for text in texts_to_predict]
  predictions = pipeline.predict(texts_to_predict)
  df['output_column'] = predictions
  return df

diff --git a/api/Neural_Network2.py b/api/Neural_Network2.py
@@ -34,9 +34,10 @@ def __init__(self, epochs):
  self.epochs = epochs
  self.epoch_num = 0
 
-
  def on_epoch_begin(self, epoch, logs=None):
  self.epoch_step = 0
+
+ def on_epoch_end(self, epoch, logs=None):
  self.epoch_num += 1
 
  def on_batch_end(self, batch, logs=None):

diff --git a/api/app.py b/api/app.py
@@ -6,13 +6,10 @@
 
 import time
 import os
-import atexit
-import threading
 import pandas as pd
 import nltk
 import json
 import asyncio
-import logging
 nltk.download('wordnet')
 
 

diff --git a/requirements.txt b/requirements.txt