import torch

from args import args

from tqdm import tqdm

from sklearn.pipeline import make_pipeline

from sklearn.preprocessing import StandardScaler

from sklearn.linear_model import LogisticRegression

def get_rep_with_label(model, dataloader):

return reps, labels

def fit_lr(features, y):

return pipe