<commit_before>from classification.model import ModelBase, TrainNetInfo

import tensorflow as tf
import tensorflow.contrib.slim as slim
import tensorflow.contrib.metrics as metrics


class Model(ModelBase):

    def build_network(self, features):

        net = slim.repeat(slim.flatten(features), 3, slim.fully_connected,
            100, activation_fn=tf.nn.relu)
        return slim.fully_connected(net, self.num_classes)

    def build_training_net(self, features, labels):
<commit_msg>ran yapf on new model<commit_after>from classification.model import ModelBase, TrainNetInfo

import tensorflow as tf
import tensorflow.contrib.slim as slim
import tensorflow.contrib.metrics as metrics


class Model(ModelBase):
    def build_network(self, features):

        net = slim.repeat(
            slim.flatten(features),
            3,
            slim.fully_connected,
            100,
            activation_fn=tf.nn.relu)
        return slim.fully_connected(net, self.num_classes)

    def build_training_net(self, features, labels):

{'fraction_non_alphanumeric': 0.07915057915057915, 'fraction_numerical': 0.007722007722007722, 'mean_word_length': 3.4506437768240343, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '2653929', 'n_tokens_mistral': 302, 'n_tokens_neox': 291, 'n_words': 75}