Antalet klasser
För att avgöra om du ska använda en binär eller en multi-klassmodell för ett klassificeringsproblem behöver du känna till antalet klasser i datan. Torch-datamängden finns tillgänglig i din arbetsyta som train_dataset.
Med tanke på antalet klasser i train_dataset – vilken typ av bildklassificering vore lämplig?
Den här övningen är en del av kursen
Djupinlärning för bilder med PyTorch
Interaktiv övning med praktiskt arbete
Gör teori till handling med en av våra interaktiva övningar
Starta övningen