s478839/ium_478839
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Zaktualizuj 'ml_pytroch_sacred.py'

Browse Source
This commit is contained in:
Sebastian Wałęsa 2022-05-08 20:33:07 +02:00
parent 8ad8722929
commit 7445be7014
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ml_pytroch_sacred.py
View File

@ -21,7 +21,7 @@ ex.observers.append(FileStorageObserver('IUM_478839'))
@ex.config @ex.config
def my_config(): def my_config():
epochs = 100 epochs = 1000
#load data #load data
dataframe = pd.read_csv("understat.csv") dataframe = pd.read_csv("understat.csv")
@ -104,7 +104,6 @@ def fit(epochs, lr, model, train_loader, val_loader, opt_func=torch.optim.SGD):
input_size = len(input_cols) input_size = len(input_cols)
output_size = len(output_cols) output_size = len(output_cols)
model=Model_xPosition() model=Model_xPosition()
epochs = 1000
lr = 1e-5 lr = 1e-5
learning_proccess = fit(epochs, lr, model, train_loader, val_loader) learning_proccess = fit(epochs, lr, model, train_loader, val_loader)
def predict_single(input, target, model): def predict_single(input, target, model):