Coder Social home page Coder Social logo

deepsmote's Introduction

DeepSMOTE

DeepSMOTE paper: https://arxiv.org/pdf/2105.02340.pdf

Code

This repository contains sample code to implement DeepSMOTE. The first file, DeepSMOTE_MNIST.py, contains code to train a model on the MNIST dataset. The second file, GenerateSamples, provides code to generate samples on a trained model.

Data and Pre-Trained Models

Sample training images and labels, as well as saved models are available for download at: https://drive.google.com/drive/folders/1GRrpiR0CJpcfpjBKO18FLjombxgqH9cK?usp=sharing

Dependencies

The code was written with: Numpy 1.17; Python 3.6; Pytorch 1.3; and Scikit learn 0.24.1.

deepsmote's People

Contributors

dd1github avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar

deepsmote's Issues

Data Structure ?

Hi, I am very interested in your repo right now. But I have some questions. In DeepSMOTE_MNIST.py file, line 178 "dtrnimg" is a txt file, why are you using os.listdir. Can you tell me the details of the data structure? Many thanks and look forward to reply from you soon.

Reconstructed Results very blurry

Hi

Thanks for adding the code for your work here.

I tried using this approach on a custom dataset where Images have more than one channel. I trained for a while and mostly the autoencoder seems to be stuck at a point where the reconstructions are just blurred images. I am not sure if it's just a matter of more training time or data, but what I noticed is that it works well when it's just reconstructing the same image using (mse) when I add the permutation part and use combo loss with the addition of (mse2), the performance deteriorates heavily and is not able to recover.

Could it be that this just doesn't work with my dataset or am I missing something.

Help is appreciated.

How to convert

The final generated TXT file, MNIST data set is a picture data set, may I ask how to convert the generated file into a picture?

Generate 0_trn_img.zip

Hi. Great work!

How did you generate 0_trn_img.txt? Is this the entire MNIST?

Cheers,

JB

Results on MNIST

Hi,

In your paper, you mentioned that DeepSMOTE could give around 96% ACSA on MNIST. Could you provide the testing scripts for that? I used your checkpoint for encoder/decoder but only get around 90% ACSA on MNIST, much lower than the number in your paper. Thank you!

Double permutation?

It seems that your second reconstruction loss is between the batch of reconstructed images (from permutated embeddings) and the permutated batch of ground truth images (in the same order!). Would that make the permutation pointless?

generate sample

hi,
i cannot run generatesample.py
it give error FileNotFoundError: [Errno 2] No such file or directory: '.../0_trn_img.txt'
where i get this file or how i can convert mnist dataset to this type of file

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.