CCS-Lab / easyml

A toolkit for easily building and evaluating machine learning models.
https://ccs-lab.github.io/easyml
Other
40 stars 16 forks source link

Train and test ROC curves show 0.50 #38

Closed paulhendricks closed 7 years ago

paulhendricks commented 7 years ago
import matplotlib as mpl; mpl.use('TkAgg')
import matplotlib.pyplot as plt
import numpy as np
import os
import pandas as pd

from easyml.glmnet import easy_glmnet

# Set current working directory
os.chdir('glmnet')

# Plot settings
plt.style.use('ggplot')

# Set seed
np.random.seed(12345)

# Load data
political_ideology = pd.read_csv('../data/political_ideology.csv')
mask = (political_ideology['y'] != 3)
political_ideology = political_ideology[mask]
political_ideology['y'] = political_ideology['y'] - 1

# Run easy_glmnet analysis

# Analyze data
easy_glmnet(political_ideology, 'y',
            family='binomial', random_state = 1, n_core = os.cpu_count(),
alpha=1, n_lambda=200, standardize=False, cut_point=0, max_iter=1e6)
paulhendricks commented 7 years ago

Closing because unable to replicate bug; must have been fixed somewhere along the way.