<commit_before>from seldon.pipeline.pandas_pipelines import BasePandasEstimator
from sklearn.base import BaseEstimator
import pandas as pd
from sklearn.utils import check_array
import tensorflow as tf

class TensorFlowWrapper(BasePandasEstimator,BaseEstimator):
        """
        Wrapper for tensorflow with pandas support
    
        Parameters
        ----------
        session : tensorflow session
            Contains the tensorflow model
        tf_input : tensorflow variable
            Variable used as the model input
        tf_output : tensorflow variable
            Variable used as the model output
        tf_constant : iterable
            List of tuples (tensorflow variable, value) to be used as constant for the model when doing predictions
        tmp_model : string
            url to folder where the model will be temporarily saved
        target : str
           Target column
        target_readable : str
           More descriptive version of target variable
        included : list str, optional
           columns to include
        excluded : list str, optional
           columns to exclude
        id_map : dict (int,str), optional
           map of class ids to high level names
        """
    def __init__(self,
                 session,<commit_msg>Small fix to tensorflow wrapper doc<commit_after>from seldon.pipeline.pandas_pipelines import BasePandasEstimator
from sklearn.base import BaseEstimator
import pandas as pd
from sklearn.utils import check_array
import tensorflow as tf

class TensorFlowWrapper(BasePandasEstimator,BaseEstimator):
    """
    Wrapper for tensorflow with pandas support

    Parameters
    ----------
    
    session : tensorflow session
        Contains the tensorflow model
    tf_input : tensorflow variable
        Variable used as the model input
    tf_output : tensorflow variable
        Variable used as the model output
    tf_constant : iterable
        List of tuples (tensorflow variable, value) to be used as constant for the model when doing predictions
    tmp_model : string
        url to folder where the model will be temporarily saved
    target : str
       Target column
    target_readable : str
       More descriptive version of target variable
    included : list str, optional
       columns to include
    excluded : list str, optional
       columns to exclude
    id_map : dict (int,str), optional
       map of class ids to high level names
    """
    def __init__(self,
                 session,
{'fraction_non_alphanumeric': 0.0392156862745098, 'fraction_numerical': 0.0, 'mean_word_length': 2.401360544217687, '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': '3086549', 'n_tokens_mistral': 604, 'n_tokens_neox': 576, 'n_words': 256}