Skip to content

Instantly share code, notes, and snippets.

@neelindresh

neelindresh/PyMLpipe - Pytorch model initiate

Created July 18, 2022 13:59
Show Gist options
  • Save neelindresh/8d391046ea6368b497882fddd607901d to your computer and use it in GitHub Desktop.
Save neelindresh/8d391046ea6368b497882fddd607901d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
PyMLpipe - Pytorch model initiate
model=Model(len(trainx.columns))
optimizer=torch.optim.SGD(model.parameters(),lr=0.001)
criterion=torch.nn.BCELoss()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment