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
Software Engineer for AI Training Data (School Specific)
@ G2i Inc | Remote
Software Engineer for AI Training Data (Python)
@ G2i Inc | Remote
Software Engineer for AI Training Data (Tier 2)
@ G2i Inc | Remote
Data Engineer
@ Lemon.io | Remote: Europe, LATAM, Canada, UK, Asia, Oceania
Artificial Intelligence – Bioinformatic Expert
@ University of Texas Medical Branch | Galveston, TX
Lead Developer (AI)
@ Cere Network | San Francisco, US