neelindresh/PyMLpipe - Pytorch model initiate

Created July 18, 2022 13:59

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/neelindresh/8d391046ea6368b497882fddd607901d.js"></script>
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