s444463/ium_444463
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

CONDA small refactoring

Browse Source
This commit is contained in:
Mikołaj Pokrywka 2022-05-23 00:09:06 +02:00
parent 955d40d1d1
commit cee02a2d2b
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
deepl.py
View File

@ -14,6 +14,8 @@ import sys
import mlflow import mlflow
from mlflow.models import infer_signature from mlflow.models import infer_signature
from urllib.parse import urlparse from urllib.parse import urlparse
import zipfile
mlflow.set_experiment("s444463x") mlflow.set_experiment("s444463x")
def convert_text_to_model_form(text): def convert_text_to_model_form(text):
@ -28,6 +30,8 @@ def train(epochs):
# print(sys.argv[1]) # print(sys.argv[1])
# epochs = int(sys.argv[1]) # epochs = int(sys.argv[1])
# epochs=10 # epochs=10
with zipfile.ZipFile('stare_zadania/real-or-fake-fake-jobposting-prediction.zip', 'r') as zip_ref:
zip_ref.extractall('')
mlflow.log_param("epochs", epochs) mlflow.log_param("epochs", epochs)
# kaggle.api.authenticate() # kaggle.api.authenticate()
@ -183,7 +187,7 @@ def train(epochs):
torch.save(model, 'model') torch.save(model, 'model')
input_example = data_train[:5] input_example = data_train[:5]
siganture = infer_signature(input_example, np.array(['company_profile'])) # siganture = infer_signature(input_example, np.array(['company_profile']))
# path = urlparse(mlflow.get_tracking_uri()).scheme # path = urlparse(mlflow.get_tracking_uri()).scheme