Commit History

Refactor DNN model to support variable number of
66f8548

ArxAlfa commited on

Update torch and pandas versions
d077299

ArxAlfa commited on

Update model weights and Docker image
9b694a7

ArxAlfa commited on

Update model hyperparameters and number of epochs
8ceaff8

ArxAlfa commited on

Update DNN architecture and KFold split
7b1ff82

ArxAlfa commited on

Update KFold n_splits and epochs
c0ad13a

ArxAlfa commited on

Update number of epochs to 100
c86fd21

ArxAlfa commited on

Refactor training loop and print average RMSE per
92e1307

ArxAlfa commited on

Update model hyperparameters and optimizer
ca246fc

ArxAlfa commited on

Update model input size in app.py
a0dbd90

ArxAlfa commited on

Fix encoding issue in app.py
5dc9f9c

ArxAlfa commited on

Fix column encoding in train function
0910d9a

ArxAlfa commited on

Update model input size and add OneHotEncoder
57ee6b1

ArxAlfa commited on

Update hidden_size parameter in model
508270a

ArxAlfa commited on

Add DNN model and training code
8f7a0cd

ArxAlfa commited on

Add model.joblib, update Dockerfile and
a650593

ArxAlfa commited on

Fix Dockerfile formatting and remove unused code
9d63bd4

ArxAlfa commited on

Update Dockerfile to use official Python 3.9 image
aad9eca

ArxAlfa commited on

Add Python 3.9 and pip installation to Dockerfile
b43c1a2

ArxAlfa commited on

Update Dockerfile to use NVIDIA CUDA image
f9b36c8

ArxAlfa commited on

Refactor translation endpoint
6f83a0d

ArxAlfa commited on

Remove unnecessary code in generate function
8da01f8

ArxAlfa commited on

Fix print statement in generate function
96f99d3

ArxAlfa commited on

Add T5Tokenizer and T5ForConditionalGeneration for
63da3d7

ArxAlfa commited on

Update translation model in app.py
3aa3dbb

ArxAlfa commited on

Update FastAPI docs and redoc URL
5d5c81e

ArxAlfa commited on

Add Dockerfile and FastAPI app with text
83bca60

ArxAlfa commited on

initial commit
d9bb0ab

ArxAlfa commited on