Changeset 16436


Ignore:
Timestamp:
Aug 1, 2016, 9:19:26 AM (3 years ago)
Author:
dferreira
Message:

Changes made to the api

Location:
internals/2016/aptoideimagesdetector/trunk
Files:
2 edited

Legend:

Unmodified
Added
Removed
  • internals/2016/aptoideimagesdetector/trunk/Source Code/analyse_app_training.py

    r16418 r16436  
    7070
    7171        featuresets = []
    72         num=0
    73         for info in app_info:
    74                 icons_l = []
    75                 screens = []
    76                 for icon in info[0]:
    77                         try:
    78                                 icons_l.append(analyse_explicit(illust2vec, icon))
    79                         except:
    80                                 print "Could not resize image"
     72        try:
     73                f = open("./featuresets.pickle", "rb")
     74                featuresets = pickle.load(f)
     75                f.close()
     76                print "Open FeatureSets Pickle"
     77        except IOError:
     78                # It can take toooo much time (4000 apps database, +- 8 hours)
     79                print "Did not Find FeatureSets Pickle"
     80                num=0
     81                for info in app_info:
     82                        icons_l = []
     83                        screens = []
     84                        for icon in info[0]:
     85                                try:
     86                                        icons_l.append(analyse_explicit(illust2vec, icon))
     87                                except:
     88                                        print "Could not resize image"
    8189
    82                 for scr in info[1]:
    83                         try:
    84                                 screens.append(analyse_explicit(illust2vec, scr))
    85                         except:
    86                                 print "Could not resize image"
     90                        for scr in info[1]:
     91                                try:
     92                                        screens.append(analyse_explicit(illust2vec, scr))
     93                                except:
     94                                        print "Could not resize image"
    8795
    88                 desc = text_cat(info[2], len(info[2]), info[3],info[4])
    89                 num+=1
    90                 if (num%100)==0:
    91                         print num
    92                 features = find_features(icons_l, screens, desc)
    93                 featuresets.append((features, info[5]))
     96                        desc = text_cat(info[2], len(info[2]), info[3],info[4])
     97                        num+=1
     98                        if (num%100)==0:
     99                                print num
     100                        features = find_features(icons_l, screens, desc)
     101                        featuresets.append((features, info[5]))
    94102
    95         random.shuffle(featuresets)
    96 
    97         save_model = open("./featuresets.pickle", "wb")
    98         pickle.dump(featuresets, save_model)
    99         save_model.close()
     103                random.shuffle(featuresets)
     104                save_model = open("./featuresets.pickle", "wb")
     105                pickle.dump(featuresets, save_model)
     106                save_model.close()
    100107       
    101108       
    102109        training_set = featuresets[:]
    103110       
    104         classifier = SklearnClassifier(SGDClassifier())
     111        classifier = SklearnClassifier(SVC(probability=True))
    105112        saving_model = classifier.train(training_set)
    106113        save_model = open("./model_apps_info.pickle", "wb")
Note: See TracChangeset for help on using the changeset viewer.