Syndria98/ith-project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

now yes

Browse Source
This commit is contained in:
BinarySandia04
2026-05-21 15:42:07 +02:00
parent e1a8d38cee
commit ba3d03b186
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
skin_cancer_classification.py
View File

@@ -271,9 +271,8 @@ x = Dense(512, activation='relu')(x) # Added another Dense layer
x = Dense(256, activation='relu')(x) # Existing Dense layer x = Dense(256, activation='relu')(x) # Existing Dense layer
predictions = Dense(1, activation='sigmoid')(x) # Output layer for binary classification predictions = Dense(1, activation='sigmoid')(x) # Output layer for binary classification
with strategy.scope(): # Use all gpus model = Model(inputs=base_model.input, outputs=predictions)
model = Model(inputs=base_model.input, outputs=predictions) model.compile(optimizer=Adam(learning_rate=0.0001), loss='binary_crossentropy', metrics=['accuracy'])
model.compile(optimizer=Adam(learning_rate=0.0001), loss='binary_crossentropy', metrics=['accuracy'])
"""## 4. Data Generators """## 4. Data Generators