def load_dataset(root_dir): try: return pickle.load(open(images_root_dir + "img_dataset.p", "rb")) except: data = np.empty((0, IMG_SIZE**2 * levels + 4), dtype=np.uint8) total = 0 for _ in glob.iglob(root_dir + '**/*.jpeg', recursive=True): total += 1 with tqdm(total=total) as pbar: for img_file in glob.iglob(images_root_dir + '**/*.jpeg', recursive=True): image = Image.open(img_file, 'r') X = np.asarray(image).reshape(-1)[ int((2640**2 * levels - IMG_SIZE**2 * levels) / 2):int((2640**2 * levels - IMG_SIZE**2 * levels) / 2 + IMG_SIZE**2 * levels)] # immagine ridotta per performace Y = get_target(img_file) data = np.append(data, np.array([np.append(X, Y)]), axis=0) pbar.update(1) pickle.dump(data, open(images_root_dir + "img_dataset.p", "wb")) return data
def write_riri_me(self, msg_text, channel_id): target = get_target(RIRI_FLAG, msg_text).upper() if target != "": txt = ' '.join(target for num in range(5)) else: txt = "WHY WOULD YOU JUST TYPE RIRI?\n" self.write_slow(txt, channel_id)
def write_riri_me(self, msg_text, channel_id): target = get_target(RIRI_FLAG, msg_text).upper() if target != "": txt = ' '.join(target for num in range(5)) else: txt = "WHY WOULD YOU JUST TYPE RIRI?\n" self.write_slow(txt, channel_id)
def write_encouragement(self, msg_text, channel_id): encouragements = [ "Get your shit together", "You can do it", "I'm here for you", "Do you just think about yourself", "You're the best", ] target = get_target(ENCOURAGE_FLAG, msg_text) txt = '{} {}'.format(random.choice(encouragements), target) self.write_slow(txt, channel_id)
def write_encouragement(self, msg_text, channel_id): encouragements = [ "Get your shit together", "You can do it", "I'm here for you", "Do you just think about yourself", "You're the best", ] target = get_target(ENCOURAGE_FLAG, msg_text) txt = '{} {}'.format(random.choice(encouragements), target) self.write_slow(txt, channel_id)
def get_house(self, msg_text): options = [] target = get_target(SORT_FLAG, msg_text).lower() if 'not' in target: for house in HOUSES: if house.lower() not in target: options.append(house) if len(options) == 0: return "" return random.choice(options) else: for house in HOUSES: if house.lower() in target: options.append(house) if len(options) == 0: options = list(HOUSES) return random.choice(options)
def create_dataset_androguard_SE(root_dir, new_dataset): total = 0 for _ in glob.iglob(root_dir + '**/*.apk', recursive=True): total += 1 data = np.empty((0,len(features_SE) + 4 + 1)) with tqdm(total=total) as pbar: for apk_file in glob.iglob(root_dir + '**/*.apk', recursive=True): try: properties = get_properties_SE(apk_file) data = np.append(data, [np.append([apk_file], np.append(properties, get_target(apk_file)))], axis=0) #data = np.append(data, [np.append([apk_file], np.append(properties, [0,1,0,0]))], axis=0) #fixed target pbar.update(1) except Exception as e: logging.error('Failed: ' + apk_file + "\t" + str(e)) targets = ['trivial', 'string', 'reflection', 'class'] df = pd.DataFrame(data=data, columns=['filename']+features_SE+targets) df.to_csv(new_dataset)
def write_sweetpotato_me(self, msg_text, channel_id): target = get_target(SWEETPOTATO_FLAG, msg_text) txt = 'Here, {}! :sweet_potato:'.format(target) self.write_slow(txt, channel_id)
def write_sass(self, msg_txt, channel_id): target = get_target(SASS_FLAG, msg_txt) sass = 'Hey, {}! {}'.format(target, self.sass_manager.get_response()) self.write_slow(sass, channel_id)
def write_french(self, msg_text, channel_id): target = get_target(FRENCH_FLAG, msg_text).replace('_', '') self.write_slow('_le {}_'.format(target), channel_id)
def write_sweetpotato_me(self, msg_text, channel_id): target = get_target(SWEETPOTATO_FLAG, msg_text) txt = 'Here, {}! :sweet_potato:'.format(target) self.write_slow(txt, channel_id)
def write_sass(self, msg_txt, channel_id): target = get_target(SASS_FLAG, msg_txt) sass = 'Hey, {}! {}'.format(target, self.sass_manager.get_response()) self.write_slow(sass, channel_id)
def write_french(self, msg_text, channel_id): target = get_target(FRENCH_FLAG, msg_text).replace('_', '') self.write_slow('_le {}_'.format(target), channel_id)