Zaktualizuj 'ml_pytroch_sacred.py'
This commit is contained in:
parent
8ad8722929
commit
7445be7014
@ -21,7 +21,7 @@ ex.observers.append(FileStorageObserver('IUM_478839'))
|
||||
|
||||
@ex.config
|
||||
def my_config():
|
||||
epochs = 100
|
||||
epochs = 1000
|
||||
|
||||
#load data
|
||||
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)
|
||||
output_size = len(output_cols)
|
||||
model=Model_xPosition()
|
||||
epochs = 1000
|
||||
lr = 1e-5
|
||||
learning_proccess = fit(epochs, lr, model, train_loader, val_loader)
|
||||
def predict_single(input, target, model):
|
||||
|
Loading…
Reference in New Issue
Block a user