all AI news
Loss is getting clamped in my GCN model [P]
Jan. 14, 2024, 5:49 p.m. | /u/Willing-Cell1790
Machine Learning www.reddit.com
class mdl(torch.nn.Module):
def init(self, input_size, hidden_size, output_size,dropout_rate):
super(GCNClassifier, self).init()
self.conv1 = GCNConv(input_size, hidden_size)
self.conv2 = GCNConv(hidden_size, output_size)
self.dropout = torch.nn.Dropout(dropout_rate)
def forward(self, x, edge_index):
x = self.conv1(x, edge_index)
x = F.relu(x)
x = self.dropout(x)
x = self.conv2(x, edge_index)
x = …
class classification data edge electronic electronic health records form graph graphs health index knowledge knowledge graph loss machinelearning node patients pytorch records torch vectors
More from www.reddit.com / Machine Learning
Jobs in AI, ML, Big Data
Senior Machine Learning Engineer
@ GPTZero | Toronto, Canada
ML/AI Engineer / NLP Expert - Custom LLM Development (x/f/m)
@ HelloBetter | Remote
Doctoral Researcher (m/f/div) in Automated Processing of Bioimages
@ Leibniz Institute for Natural Product Research and Infection Biology (Leibniz-HKI) | Jena
Seeking Developers and Engineers for AI T-Shirt Generator Project
@ Chevon Hicks | Remote
Principal Data Architect - Azure & Big Data
@ MGM Resorts International | Home Office - US, NV
GN SONG MT Market Research Data Analyst 11
@ Accenture | Bengaluru, BDC7A