https://github.com/unonao/atmacup-15
CV
Feature
Model
layers = []
layers.append(nn.Linear(n_features, hidden_dim))
layers.append(nn.ReLU())
layers.append(nn.Dropout(p=dropout_rate))
for _ in range(n_layers - 2):
layers.append(nn.Linear(hidden_dim, hidden_dim))
layers.append(nn.ReLU())
layers.append(nn.Dropout(p=dropout_rate)) # Dropout層を追加
self.fc_layers = nn.Sequential(*layers)
self.fc_final = nn.Linear(hidden_dim, 1)
実験管理