Advertisement
Guest User

Untitled

a guest
Nov 13th, 2019
876
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.55 KB | None | 0 0
  1. import pandas as pd
  2. from sklearn.model_selection import train_test_split
  3. from sklearn.linear_model import LogisticRegression
  4.  
  5. data = pd.read_csv('/datasets/travel_insurance.csv')
  6.  
  7. data_ohe = pd.get_dummies(data, drop_first=True)
  8. target = data_ohe['Claim']
  9. features = data_ohe.drop('Claim', axis=1)
  10.  
  11. features_train, features_valid, target_train, target_valid = train_test_split(features, target, test_size=0.25, random_state=12345)
  12.  
  13. model = LogisticRegression(random_state=12345)
  14.  
  15. model.fit(features_train, target_train)
  16.  
  17. print("Обучено!")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement